Fixing model calling in graph.py

graph.py

@@ -35,7 +35,7 @@ def call_model(state: AgentState) -> AgentState:
     """
     try:
         # Access the model from the global context
-        model = get_model()
+        model = call_model.__globals__["_model"]
         messages = state.messages
         response = model.invoke(messages)
         return {"messages": [response]}