diff --git a/llm_hello_world.py b/llm_hello_world.py deleted file mode 100644 index 6b42ef5ef..000000000 --- a/llm_hello_world.py +++ /dev/null @@ -1,31 +0,0 @@ -#!/usr/bin/env python -# -*- coding: utf-8 -*- -""" -@Time : 2023/5/6 14:13 -@Author : alexanderwu -@File : llm_hello_world.py -""" -import asyncio - -from metagpt.logs import logger -from metagpt.llm import LLM -from metagpt.llm import Claude - -async def main(): - llm = LLM() - claude = Claude() - logger.info(await claude.aask('你好,请进行自我介绍')) - logger.info(await llm.aask('hello world')) - logger.info(await llm.aask_batch(['hi', 'write python hello world.'])) - - hello_msg = [{'role': 'user', 'content': 'count from 1 to 10. split by newline.'}] - logger.info(await llm.acompletion(hello_msg)) - logger.info(await llm.acompletion_batch([hello_msg])) - logger.info(await llm.acompletion_batch_text([hello_msg])) - - logger.info(await llm.acompletion_text(hello_msg)) - await llm.acompletion_text(hello_msg, stream=True) - - -if __name__ == '__main__': - asyncio.run(main()) diff --git a/metagpt/provider/anthropic_api.py b/metagpt/provider/anthropic_api.py index fae4e53eb..36cd8e27a 100644 --- a/metagpt/provider/anthropic_api.py +++ b/metagpt/provider/anthropic_api.py @@ -13,7 +13,7 @@ from metagpt.config import CONFIG class Claude2: def ask(self, prompt): - client = Anthropic(api_key=claude_api_key) + client = Anthropic(api_key=CONFIG.claude_api_key) res = client.completions.create( model="claude-2", @@ -23,7 +23,7 @@ class Claude2: return res.completion async def aask(self, prompt): - client = Anthropic(api_key=claude_api_key) + client = Anthropic(api_key=CONFIG.claude_api_key) res = client.completions.create( model="claude-2",