Upload 2 files

app.py

@@ -22,28 +22,18 @@ def setup_client(provider):
         if not api_key:
             st.error("OPENAI_API_KEY not set in environment variables.")
             return None, None, None
-        client = instructor.from_openai(AsyncOpenAI(api_key=api_key))
+        # Explicitly pass only the required arguments to AsyncOpenAI
+        openai_client = AsyncOpenAI(api_key=api_key)
+        client = instructor.from_openai(openai_client)
         model = "gpt-4o-mini"
         display_model = "OpenAI (gpt-4o-mini)"
     elif provider == "ollama":
         from openai import AsyncOpenAI as OllamaClient
-        client = instructor.from_openai(
-            OllamaClient(base_url="http://localhost:11434/v1", api_key="ollama"), mode=instructor.Mode.JSON
-        )
+        # Explicitly pass only the required arguments to OllamaClient
+        ollama_client = OllamaClient(base_url="http://localhost:11434/v1", api_key="ollama")
+        client = instructor.from_openai(ollama_client, mode=instructor.Mode.JSON)
         model = "llama3"
         display_model = "Ollama (llama3)"
-    # elif provider == "gemini":
-    #     from openai import AsyncOpenAI
-    #     api_key = os.getenv("GEMINI_API_KEY")
-    #     if not api_key:
-    #         st.error("GEMINI_API_KEY not set in environment variables.")
-    #         return None, None, None
-    #     client = instructor.from_openai(
-    #         AsyncOpenAI(api_key=api_key, base_url="https://generativelanguage.googleapis.com/v1beta/openai/"),
-    #         mode=instructor.Mode.JSON,
-    #     )
-    #     model = "gemini-2.0-flash-exp"
-    #     display_model = "Gemini (gemini-2.0-flash-exp)"
     else:
         st.error(f"Unsupported provider: {provider}")
         return None, None, None

requirements.txt

@@ -1,4 +1,5 @@
 streamlit==1.38.0
 instructor==1.3.4
 atomic-agents==0.2.1
-python-dotenv==1.0.1
+python-dotenv==1.0.1
+openai==1.40.0