From 326082da728a3d3c6cfedabf0f68cb626f04b857 Mon Sep 17 00:00:00 2001
From: Leo <xiaolingfeng@fuzhi.ai>
Date: Fri, 21 Jul 2023 13:53:39 -0400
Subject: [PATCH] Add Anthropic Api

---
 llm_hello_world.py                | 31 -------------------------------
 metagpt/provider/anthropic_api.py |  4 ++--
 2 files changed, 2 insertions(+), 33 deletions(-)
 delete mode 100644 llm_hello_world.py

diff --git a/llm_hello_world.py b/llm_hello_world.py
deleted file mode 100644
index 6b42ef5ef..000000000
--- a/llm_hello_world.py
+++ /dev/null
@@ -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())
diff --git a/metagpt/provider/anthropic_api.py b/metagpt/provider/anthropic_api.py
index fae4e53eb..36cd8e27a 100644
--- a/metagpt/provider/anthropic_api.py
+++ b/metagpt/provider/anthropic_api.py
@@ -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",