From 832294809b097793dff3472b1183aed37f8f5c8d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Thu, 7 Sep 2023 11:01:27 +0800
Subject: [PATCH] feat: + LLMType

---
 metagpt/llm.py               | 24 ++++++++++++++++++++++--
 metagpt/provider/__init__.py |  4 +++-
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/metagpt/llm.py b/metagpt/llm.py
index 6a9a9132f..0ef23d0be 100644
--- a/metagpt/llm.py
+++ b/metagpt/llm.py
@@ -4,17 +4,37 @@
 @Time    : 2023/5/11 14:45
 @Author  : alexanderwu
 @File    : llm.py
+@Modified By: mashenquan, 2023
 """
+from enum import Enum
 
 from metagpt.provider.anthropic_api import Claude2 as Claude
 from metagpt.provider.openai_api import OpenAIGPTAPI as LLM
 
+
+class LLMType(Enum):
+    OPENAI = "OpenAI"
+    METAGPT = "MetaGPT"
+    UNKNOWN = "UNKNOWN"
+
+    @classmethod
+    def get(cls, value):
+        for member in cls:
+            if member.value == value:
+                return member
+        return cls.UNKNOWN
+
+    @property
+    def UNKNOWN(self):
+        return LLMType.UNKNOWN
+
+
 DEFAULT_LLM = LLM()
 CLAUDE_LLM = Claude()
 
 
 async def ai_func(prompt):
     """使用LLM进行QA
-       QA with LLMs
-     """
+    QA with LLMs
+    """
     return await DEFAULT_LLM.aask(prompt)
diff --git a/metagpt/provider/__init__.py b/metagpt/provider/__init__.py
index 56dc19b4b..9895aa7fc 100644
--- a/metagpt/provider/__init__.py
+++ b/metagpt/provider/__init__.py
@@ -4,9 +4,11 @@
 @Time    : 2023/5/5 22:59
 @Author  : alexanderwu
 @File    : __init__.py
+@Modified By: mashenquan, 2023/9/8. Add `MetaGPTLLMAPI`
 """
 
 from metagpt.provider.openai_api import OpenAIGPTAPI
+from metagpt.provider.metagpt_llm_api import MetaGPTLLMAPI
 
 
-__all__ = ["OpenAIGPTAPI"]
+__all__ = ["OpenAIGPTAPI", "MetaGPTLLMAPI"]