logic refine

ufo/agent/basic.py

@@ -4,12 +4,11 @@
 import json
 from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
-from typing import Dict, List, Type, Optional
+from typing import Dict, List, Optional, Type
 
 from ufo import utils
-from ufo.module.state import Status
 from ufo.llm import llm_call
-
+from ufo.module.state import Status
 
 # Lazy import the retriever factory to aviod long loading time.
 retriever = utils.LazyImport("..rag.retriever")
@@ -251,7 +250,9 @@ def message_constructor(self) -> List[dict]:
         pass
 
     @classmethod
-    def get_response(cls, message: List[dict], namescope, use_backup_engine) -> str:
+    def get_response(
+        cls, message: List[dict], namescope: str, use_backup_engine: bool
+    ) -> str:
         """
         Get the response for the prompt.
         :param prompt: The prompt.

ufo/llm/openai.py

@@ -11,7 +11,16 @@
 
 
 class OpenAIService(BaseService):
-    def __init__(self, config, agent_type: str):
+    """
+    The OpenAI service class to interact with the OpenAI API.
+    """
+
+    def __init__(self, config, agent_type: str) -> None:
+        """
+        Create an OpenAI service instance.
+        :param config: The configuration for the OpenAI service.
+        :param agent_type: The type of the agent.
+        """
         self.config_llm = config[agent_type]
         self.config = config
         self.api_type = self.config_llm["API_TYPE"].lower()

ufo/module/processors/processor.py

@@ -56,15 +56,15 @@ def __init__(
         """
 
         super().__init__(
-            round_num,
-            log_path,
-            request,
-            request_logger,
-            logger,
-            round_step,
-            session_step,
-            prev_status,
-            app_window,
+            round_num=round_num,
+            log_path=log_path,
+            request=request,
+            request_logger=request_logger,
+            logger=logger,
+            round_step=round_step,
+            session_step=session_step,
+            prev_status=prev_status,
+            app_window=app_window,
         )
 
         self.host_agent = host_agent
@@ -419,15 +419,15 @@ def __init__(
         """
 
         super().__init__(
-            round_num,
-            log_path,
-            request,
-            request_logger,
-            logger,
-            round_step,
-            session_step,
-            prev_status,
-            app_window,
+            round_num=round_num,
+            log_path=log_path,
+            request=request,
+            request_logger=request_logger,
+            logger=logger,
+            round_step=round_step,
+            session_step=session_step,
+            prev_status=prev_status,
+            app_window=app_window,
         )
 
         self.app_agent = app_agent
@@ -774,7 +774,7 @@ def update_status(self) -> None:
         if self._status != Status.FINISH:
             time.sleep(configs["SLEEP_TIME"])
 
-    def _safe_guard_judgement(self, action, control_text) -> bool:
+    def _safe_guard_judgement(self, action: str, control_text: str) -> bool:
         """
         Safe guard for the session.
         action: The action to be taken.