Update agent

GnosisPages.py

@@ -3,16 +3,15 @@
 import os
 import streamlit as st
 import openai
-from dotenv import load_dotenv
 from gnosis.chroma_client import ChromaDB
 import gnosis.gui_messages as gm
 from gnosis import settings
 from gnosis.components.sidebar import sidebar
 from gnosis.components.main import main
+from pathlib import Path
+from dotenv import load_dotenv
 
-
-load_dotenv()
-
+load_dotenv(Path(__file__).parent / ".env")
 
 openai.api_key = os.getenv("OPENAI_API_KEY")
 

gnosis/agent.py

@@ -14,7 +14,7 @@ class PDFExplainer:
         self.tools = [
             Tool.from_function(
                 func=search.run,
-                name="Search on ChromaDB",
+                name="Search_on_ChromaDB",
                 description="Useful when you need more context for answering a question.",
                 handle_parsing_errors=True,
             )
@@ -26,7 +26,8 @@ class PDFExplainer:
         self.agent = initialize_agent(
             self.tools,
             llm,
-            agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+            #agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+            agent=AgentType.OPENAI_FUNCTIONS,  # ← uses OpenAI's native function calling
             verbose=False,
             handle_parsing_errors=True,
         )

gnosis/builder.py

@@ -1,24 +1,37 @@
 """Module for building the Langchain Agent"""
+import os
 import streamlit as st
 from langchain_openai import ChatOpenAI
 from gnosis.agent import PDFExplainer
+from openai import AuthenticationError, APIConnectionError, BadRequestError
 
 
 def build(key, client):
     """An Agent builder"""
     # Build Agent
     try:
+        key = os.environ.get("OPENAI_API_KEY") or key
         llm = ChatOpenAI(
             temperature=st.session_state.temperature,
             model="gpt-4o-mini",
-            api_key=key,
-        )
+            api_key=key
+            )
         agent = PDFExplainer(
             llm,
             client,
             extra_tools=st.session_state.wk_button,
         ).agent
-    except Exception:  # pylint: disable=broad-exception-caught
-        st.warning("Missing OpenAI API Key.")
+    except Exception as e:  # pylint: disable=broad-exception-caught
+        st.warning(f"Error building agent: {e}")
+    except AuthenticationError:
+        st.warning("Invalid OpenAI API Key. Check your credentials.")
+    except APIConnectionError:
+        st.warning("Could not connect to OpenAI. Check your network connection.")
+    except BadRequestError as e:
+        st.warning(f"Bad request: {e}")
+    except ValueError as e:
+        st.warning(f"Configuration error: {e}")
+    except Exception as e:  # pylint: disable=broad-exception-caught
+        st.warning(f"Unexpected error building agent: {e}")
 
     return agent