update

Canner · Dec 27, 2024 · ac208ce · ac208ce
1 parent 6635bfb
commit ac208ce
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/wren-ai-service/src/pipelines/generation/user_guide_assistance.py b/wren-ai-service/src/pipelines/generation/user_guide_assistance.py
@@ -105,6 +105,8 @@ class UserGuideAssistance(BasicPipeline):
     def __init__(
         self,
         llm_provider: LLMProvider,
+        is_oss: bool,
+        doc_endpoint: str,
         **kwargs,
     ):
         self._user_queues = {}
@@ -119,8 +121,8 @@ def __init__(
             ),
         }
         self._configs = {
-            "is_oss": kwargs.get("is_oss", True),
-            "doc_endpoint": kwargs.get("doc_endpoint", "https://docs.getwren.ai"),
+            "is_oss": is_oss,
+            "doc_endpoint": doc_endpoint,
         }
 
         super().__init__(