update

agent.py

@@ -1,6 +1,8 @@
 import operator
 import os
 import time
+from typing import Optional
+
 from langchain.chat_models import init_chat_model
 from langchain_core.messages import HumanMessage, SystemMessage, AIMessage, AnyMessage
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
@@ -15,8 +17,8 @@ from typing_extensions import TypedDict, Annotated
 
 class State(TypedDict):
     messages: Annotated[list, add_messages]
-    content_type: str
-    content: str
+    content_type: Optional[str]
+    content: Optional[str]
     aggregate: Annotated[list, operator.add]
     # graph_state: str
 
@@ -85,7 +87,8 @@ def get_graph(llm):
     def make_plan(state: State):
 
         print("\n-------------------- Starting to create a plan --------------------\n")
-        print("Content is: ", state["content_type"])
+        if "content_type" in state:
+            print("Content is: ", state["content"])
         # get all messages from the state
         messages = state["messages"]
         # append planning message

app.py

@@ -82,8 +82,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     # 2. Fetch Questions
     print(f"Fetching questions from: {questions_url}")
     try:
-        #response = requests.get(questions_url, timeout=15)
-        response = requests.get("https://agents-course-unit4-scoring.hf.space/questions", timeout=20)
+        response = requests.get(questions_url, timeout=15)
+        #response = requests.get("https://agents-course-unit4-scoring.hf.space/questions", timeout=20)
         response.raise_for_status()
         questions_data = response.json()
         #with open('data.json', 'r') as json_file: