Dhruv-Ty commited on
Commit
d65a2ca
·
1 Parent(s): 44d5f7f
Files changed (1) hide show
  1. app.py +20 -6
app.py CHANGED
@@ -5,7 +5,6 @@ from dotenv import load_dotenv
5
  from transformers import logging
6
 
7
  from langgraph.checkpoint.memory import MemorySaver
8
- # from langchain_community.chat_models import ChatOpenRouter
9
  from langchain_openai import ChatOpenAI
10
 
11
  from interface import create_demo
@@ -15,7 +14,20 @@ from medrax.utils import *
15
 
16
  warnings.filterwarnings("ignore")
17
  logging.set_verbosity_error()
18
- _ = load_dotenv()
 
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  def initialize_agent(
21
  prompt_file,
@@ -57,10 +69,12 @@ def initialize_agent(
57
  tools_dict[tool_name] = all_tools[tool_name]()
58
 
59
  checkpointer = MemorySaver()
 
 
60
  model = ChatOpenAI(
61
- model_name=model, # "google/gemini-2.5-pro-exp-03-25:free"
62
- api_key=os.getenv("OPENAI_API_KEY"),
63
- base_url=os.getenv("OPENAI_BASE_URL"),
64
  temperature=temperature,
65
  top_p=top_p,
66
  )
@@ -105,4 +119,4 @@ if __name__ == "__main__":
105
  )
106
 
107
  demo = create_demo(agent, tools_dict)
108
- demo.launch(server_name="0.0.0.0", server_port=8585, share=True)
 
5
  from transformers import logging
6
 
7
  from langgraph.checkpoint.memory import MemorySaver
 
8
  from langchain_openai import ChatOpenAI
9
 
10
  from interface import create_demo
 
14
 
15
  warnings.filterwarnings("ignore")
16
  logging.set_verbosity_error()
17
+ load_dotenv() # Remove the _ = assignment to ensure it's properly loaded
18
+
19
+ # Set environment variables explicitly to ensure they're available
20
+ api_key = os.getenv("OPENAI_API_KEY")
21
+ base_url = os.getenv("OPENAI_BASE_URL")
22
+
23
+ if not api_key:
24
+ raise ValueError("OPENAI_API_KEY not found in environment variables")
25
+ if not base_url:
26
+ raise ValueError("OPENAI_BASE_URL not found in environment variables")
27
+
28
+ # Set them in environment for libraries that might read directly from os.environ
29
+ os.environ["OPENAI_API_KEY"] = api_key
30
+ os.environ["OPENAI_BASE_URL"] = base_url
31
 
32
  def initialize_agent(
33
  prompt_file,
 
69
  tools_dict[tool_name] = all_tools[tool_name]()
70
 
71
  checkpointer = MemorySaver()
72
+
73
+ # Explicitly pass the API key and base URL
74
  model = ChatOpenAI(
75
+ model_name=model,
76
+ api_key=api_key,
77
+ base_url=base_url,
78
  temperature=temperature,
79
  top_p=top_p,
80
  )
 
119
  )
120
 
121
  demo = create_demo(agent, tools_dict)
122
+ demo.launch(server_name="0.0.0.0", server_port=8585, share=True)