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)
|