apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-30 14:35:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Anthropic Api

Browse source
This commit is contained in:
Leo 2023-07-21 13:53:39 -04:00
parent 227c837d0a
commit 326082da72
2 changed files with 2 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

31
llm_hello_world.py
View file

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

4
metagpt/provider/anthropic_api.py
View file

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