From 3a4f31b51787f2e60bed4efda45f63d49e1637ec 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 12:00:30 +0800
Subject: [PATCH] feat: +LLMType

---
 metagpt/actions/action.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/metagpt/actions/action.py b/metagpt/actions/action.py
index e4b9613ad..c52caaa40 100644
--- a/metagpt/actions/action.py
+++ b/metagpt/actions/action.py
@@ -5,6 +5,7 @@
 @Author  : alexanderwu
 @File    : action.py
 @Modified By: mashenquan, 2023/8/20. Add function return annotations.
+@Modified By: mashenquan, 2023/9/8. Replace LLM with LLMFactory
 """
 from __future__ import annotations
 
@@ -14,16 +15,17 @@ from typing import Optional
 from tenacity import retry, stop_after_attempt, wait_fixed
 
 from metagpt.actions.action_output import ActionOutput
-from metagpt.llm import LLM
+from metagpt.llm import LLMFactory
 from metagpt.logs import logger
+from metagpt.provider.base_gpt_api import BaseGPTAPI
 from metagpt.utils.common import OutputParser
 
 
 class Action(ABC):
-    def __init__(self, name: str = "", context=None, llm: LLM = None):
+    def __init__(self, name: str = "", context=None, llm: BaseGPTAPI = None):
         self.name: str = name
         if llm is None:
-            llm = LLM()
+            llm = LLMFactory.new_llm()
         self.llm = llm
         self.context = context
         self.prefix = ""