fix changes

core/helper.py

@@ -4,28 +4,28 @@ from core.lifecycle import Lifecycle
 class LifecycleHelper:
     @classmethod
     def initialize_if_possible(cls, ls: Lifecycle) -> None:
-        if isinstance(ls, Lifecycle) and ls.get_lifecycle_state().can_initialize(
+        if isinstance(ls, Lifecycle) and ls.lifecycle_state.can_initialize(
             ls.lifecycle_state.phase
         ):
             ls.initialize()
 
     @classmethod
     def start_if_possible(cls, ls: Lifecycle) -> None:
-        if isinstance(ls, Lifecycle) and ls.get_lifecycle_state().can_start(
+        if isinstance(ls, Lifecycle) and ls.lifecycle_state.can_start(
             ls.lifecycle_state.phase
         ):
             ls.start()
 
     @classmethod
     def stop_if_possible(cls, ls: Lifecycle) -> None:
-        if isinstance(ls, Lifecycle) and ls.get_lifecycle_state().can_stop(
+        if isinstance(ls, Lifecycle) and ls.lifecycle_state.can_stop(
             ls.lifecycle_state.phase
         ):
             ls.stop()
 
     @classmethod
     def dispose_if_possible(cls, ls: Lifecycle) -> None:
-        if isinstance(ls, Lifecycle) and ls.get_lifecycle_state().can_dispose(
+        if isinstance(ls, Lifecycle) and ls.lifecycle_state.can_dispose(
             ls.lifecycle_state.phase
         ):
             ls.dispose()

langchain_manager/manager.py

@@ -24,16 +24,18 @@ class BaseLangChainManager(ABC):
 class LangChainAzureManager(BaseLangChainManager):
     def __init__(self) -> None:
         super().__init__()
+        self.embedding = OpenAIEmbeddings(client=None, chunk_size=1)
+        self.llm = AzureOpenAI(
+            deployment_name="text-davinci-003",
+            # model_name="text-davinci-003",
+            model="text-davinci-003",
+            client=None,
+        )
 
     # Override
     def get_embedding(self) -> LCEmbeddings:
-        return OpenAIEmbeddings(client=None, chunk_size=1)
+        return self.embedding
 
     # Override
     def get_llm(self) -> BaseLanguageModel:
-        return AzureOpenAI(
-            deployment_name="text-davinci-003",
-            # model_name="text-davinci-003",
-            model="text-davinci-003",
-            client=None,
-        )
+        return self.llm

xpipe_wiki/manager_factory.py

@@ -9,20 +9,22 @@ class XPipeRobotRevision(enum.Enum):
     SIMPLE_OPENAI_VERSION_0 = 1
 
 
-CAPABLE = dict()
-"""
-CAPABLE: Dict[XPipeRobotRevision, XPipeWikiRobotManager] = {XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0: XPipeWikiRobotManager()}
-"""
+
+
 
 
 class XPipeRobotManagerFactory:
+    """
+    CAPABLE: Dict[XPipeRobotRevision, XPipeWikiRobotManager] = {XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0: XPipeWikiRobotManager()}
+    """
+    CAPABLE = dict()
     @classmethod
     def get_or_create(cls, revision: XPipeRobotRevision) -> XPipeWikiRobotManager:
-        if CAPABLE.get(revision) is not None:
-            return CAPABLE[revision]
+        if cls.CAPABLE.get(revision) is not None:
+            return cls.CAPABLE[revision]
         if revision == XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0:
             manager = cls.create_simple_openai_version_0()
-        CAPABLE[revision] = manager
+        cls.CAPABLE[revision] = manager
         return manager
 
     @classmethod
@@ -48,3 +50,4 @@ class XPipeRobotManagerFactory:
         LifecycleHelper.initialize_if_possible(robot_manager)
         LifecycleHelper.start_if_possible(robot_manager)
         return robot_manager
+

xpipe_wiki/robot_manager.py

@@ -36,6 +36,7 @@ class XPipeWikiRobotManager(Lifecycle):
 class AzureXPipeWikiRobotManager(XPipeWikiRobotManager):
     service_context_manager: ServiceContextManager
     storage_context_manager: StorageContextManager
+    query_engine: BaseQueryEngine
 
     def __init__(
         self,
@@ -47,13 +48,7 @@ class AzureXPipeWikiRobotManager(XPipeWikiRobotManager):
         self.storage_context_manager = storage_context_manager
 
     def get_robot(self) -> XPipeWikiRobot:
-        index = load_index_from_storage(
-            storage_context=self.storage_context_manager.get_storage_context()
-        )
-        query_engine = index.as_query_engine(
-            service_context=self.service_context_manager.get_service_context()
-        )
-        return AzureOpenAIXPipeWikiRobot(query_engine)
+        return AzureOpenAIXPipeWikiRobot(self.query_engine)
 
     def do_init(self) -> None:
         LifecycleHelper.initialize_if_possible(self.service_context_manager)
@@ -62,6 +57,12 @@ class AzureXPipeWikiRobotManager(XPipeWikiRobotManager):
     def do_start(self) -> None:
         LifecycleHelper.start_if_possible(self.service_context_manager)
         LifecycleHelper.start_if_possible(self.storage_context_manager)
+        index = load_index_from_storage(
+            storage_context=self.storage_context_manager.get_storage_context()
+        )
+        self.query_engine = index.as_query_engine(
+            service_context=self.service_context_manager.get_service_context()
+        )
 
     def do_stop(self) -> None:
         LifecycleHelper.stop_if_possible(self.storage_context_manager)