From 227c837d0a26b9a1bb5c4ece322aa8ae051c8d22 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=82=96=E5=87=8C=E9=A3=8E?= Date: Fri, 21 Jul 2023 13:44:58 -0400 Subject: [PATCH] Your commit message --- llm_hello_world.py | 31 +++++++++++++++++++++++++++++++ metagpt/provider/anthropic_api.py | 2 +- 2 files changed, 32 insertions(+), 1 deletion(-) create mode 100644 llm_hello_world.py 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",