diff --git a/llm_hello_world.py b/llm_hello_world.py new file mode 100644 index 000000000..6b42ef5ef --- /dev/null +++ b/llm_hello_world.py @@ -0,0 +1,31 @@ +#!/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 ef1c21188..fae4e53eb 100644 --- a/metagpt/provider/anthropic_api.py +++ b/metagpt/provider/anthropic_api.py @@ -23,7 +23,7 @@ class Claude2: return res.completion async def aask(self, prompt): - client = Anthropic(api_key="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") + client = Anthropic(api_key=claude_api_key) res = client.completions.create( model="claude-2",