🚑️ add logging for llama-index openai and add default model_name

alan-turing-institute · Sep 27, 2023 · 70690bf · 70690bf
1 parent a7a27ce
commit 70690bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/reginald/models/models/__init__.py b/reginald/models/models/__init__.py
@@ -29,7 +29,7 @@
     "llama-index-llama-cpp": "https://huggingface.co/TheBloke/Llama-2-13B-chat-GGUF/resolve/main/llama-2-13b-chat.Q6_K.gguf",
     "llama-index-hf": "StabilityAI/stablelm-tuned-alpha-3b",
     "llama-index-gpt-azure": "reginald-gpt35-turbo",
-    "llama-index-gpt-openai": None,
+    "llama-index-gpt-openai": "gpt-3.5-turbo",
 }
 
 __all__ = ["MODELS", "DEFAULTS"]
diff --git a/reginald/models/models/llama_index.py b/reginald/models/models/llama_index.py
@@ -667,6 +667,7 @@ def __init__(
         super().__init__(*args, model_name=self.model_name, **kwargs)
 
     def _prep_llm(self) -> LLM:
+        logging.info(f"Setting up OpenAI LLM (model {self.model_name})")
         return OpenAI(
             model=self.model_name,
             temperature=self.temperature,