fix error in llama-index-azure-gpt model getter

alan-turing-institute · Jun 12, 2024 · 20a3eb5 · 20a3eb5
1 parent 0d2737b
commit 20a3eb5
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,7 +25,7 @@ httpx = "^0.27.0"
 llama-cpp-python = "^0.2.78"
 llama-index = "^0.10.43"
 nest_asyncio = "^1.6.0"
-openai = "^1.33.0"
+openai = "^1.34.0"
 pandas = "^2.2.2"
 pulumi = { version="^3.100.0", optional=true }
 pulumi-azure-native = { version="^2.24.0", optional=true }

diff --git a/reginald/models/__init__.py b/reginald/models/__init__.py
@@ -44,8 +44,8 @@ def get_model(model_name: str):
 
                 return LlamaIndexHF
             case "llama-index-gpt-azure":
-                from reginald.models.llama_index.llama_index_openai import (
-                    LlamaIndexGPTOpenAI,
+                from reginald.models.llama_index.llama_index_azure_openai import (
+                    LlamaIndexGPTAzure,
                 )
 
                 return LlamaIndexGPTOpenAI

diff --git a/reginald/models/chat_interact.py b/reginald/models/chat_interact.py
@@ -25,6 +25,8 @@ def run_chat_interact(streaming: bool = False, **kwargs) -> ResponseModel:
         message = input(">>> ")
         if message in ["exit", "exit()", "quit()", "bye Reginald"]:
             return response_model
+        if message == "":
+            continue
         if message in ["clear_history", "\clear_history"]:
             if (
                 response_model.mode == "chat"