Your commit message

This commit is contained in:
肖凌风 2023-07-21 13:44:58 -04:00
parent 8c07be0e56
commit 227c837d0a
2 changed files with 32 additions and 1 deletions

31
llm_hello_world.py Normal file
View file

@ -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())

View file

@ -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",