add comments

Canner · Jul 1, 2024 · 7adc61c · 7adc61c
1 parent 8426b26
commit 7adc61c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/wren-ai-service/src/providers/llm/azure_openai.py b/wren-ai-service/src/providers/llm/azure_openai.py
@@ -136,6 +136,9 @@ def get_generator(
         or GENERATION_MODEL_KWARGS,
         system_prompt: Optional[str] = None,
     ):
+        logger.info(
+            f"Creating Azure OpenAI generator with model kwargs: {model_kwargs}"
+        )
         return AsyncGenerator(
             api_key=self._generation_api_key,
             model=self._generation_model,

diff --git a/wren-ai-service/src/providers/llm/ollama.py b/wren-ai-service/src/providers/llm/ollama.py
@@ -138,6 +138,7 @@ def get_generator(
         or GENERATION_MODEL_KWARGS,
         system_prompt: Optional[str] = None,
     ):
+        logger.info(f"Creating Ollama generator with model kwargs: {model_kwargs}")
         return AsyncGenerator(
             model=self._generation_model,
             url=f"{self._url}/api/generate",

diff --git a/wren-ai-service/src/providers/llm/openai.py b/wren-ai-service/src/providers/llm/openai.py
@@ -146,6 +146,13 @@ def get_generator(
         or GENERATION_MODEL_KWARGS,
         system_prompt: Optional[str] = None,
     ):
+        if self._api_base == LLM_OPENAI_API_BASE:
+            logger.info(f"Creating OpenAI generator with model kwargs: {model_kwargs}")
+        else:
+            logger.info(
+                f"Creating OpenAI API-compatible generator with model kwargs: {model_kwargs}"
+            )
+
         return AsyncGenerator(
             api_key=self._api_key,
             api_base_url=self._api_base,