Skip to content

Commit

Permalink
Your commit message
Browse files Browse the repository at this point in the history
  • Loading branch information
肖凌风 committed Jul 21, 2023
1 parent 8c07be0 commit 227c837
Show file tree
Hide file tree
Showing 2 changed files with 32 additions and 1 deletion.
31 changes: 31 additions & 0 deletions llm_hello_world.py
Original file line number Diff line number Diff line change
@@ -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())
2 changes: 1 addition & 1 deletion metagpt/provider/anthropic_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ def ask(self, prompt):
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",
Expand Down

0 comments on commit 227c837

Please sign in to comment.