fix
Browse files
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 |
-
_ =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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,
|
62 |
-
api_key=
|
63 |
-
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)
|