add observability

.gitignore

@@ -10,4 +10,3 @@ wheels/
 .venv
 *.lock
 .env
-.python_version

.huggingface/space.yaml

@@ -1 +0,0 @@
-sdk: docker

agent.py

@@ -1,3 +1,4 @@
+import logfire
 from deep_research_agent.nodes import (
     BeginResearch,
     FinalReport,
@@ -8,6 +9,9 @@ from deep_research_agent.nodes import (
 from deep_research_agent.state import ResearchState
 from pydantic_graph import Graph
 
+logfire.configure()
+logfire.instrument_pydantic_ai()
+
 
 async def query_agent(research_query: str):
     state = ResearchState()

app.py

@@ -1,18 +1,9 @@
 import gradio as gr
-import asyncio
 from agent import query_agent
 
 
 # Simulate your async agent interaction
 async def interact_with_agent(prompt: str):
-    # # This is where you'd call your real query_agent()
-    # await asyncio.sleep(1)
-    # urls = [
-    #     "https://arxiv.org/abs/1234.5678",
-    #     "https://openai.com/research",
-    #     "https://example.com/article",
-    # ]
-    # report = f"Report generated for: '{prompt}'"
     final_report = await query_agent(research_query=prompt)
     final_report, refs = (
         final_report.split("\n\n## References:\n\n")[0],
@@ -28,35 +19,50 @@ def format_urls(urls):
     return urls
 
 
-with gr.Blocks(title="Deep Research Agent") as chat_app:
-    gr.Markdown("## 🧠 Deep Research Agent")
+def start_new_query(message):
+    # Reset chat and sidebar
+    return message, [], ""
 
-    with gr.Row():
-        # Main chat area
-        with gr.Column(scale=4):
-            chatbot = gr.Chatbot(type="messages", label="Chat", height=550)
-            msg = gr.Textbox(placeholder="Ask a research question...")
-            clear = gr.ClearButton([msg, chatbot])
 
-        # Sidebar for URLs
-        with gr.Column(scale=1):
-            gr.Markdown("### 🔗 References")
-            url_display = gr.Markdown("(No sources yet)")
+if __name__ == "__main__":
+    with gr.Blocks(title="Deep Research Agent", fill_height=True) as chat_app:
+        gr.Markdown("## 🧠 Deep Research Agent")
 
-    # Step 1: show user message immediately
-    def show_user_msg(message, chat_history: list):
-        chat_history.append({"role": "user", "content": message})
-        return chat_history
+        with gr.Row():
+            # Main chat area
+            with gr.Column(scale=4):
+                chatbot = gr.Chatbot(type="messages", label="Chat", height=550)
+                msg = gr.Textbox(placeholder="Ask a research question...")
+                clear = gr.ClearButton([msg, chatbot])
 
-    # Step 2: respond asynchronously (simulate the agent)
-    async def respond(message: str, chat_history: list):
-        report, urls = await interact_with_agent(message)
-        chat_history.append({"role": "assistant", "content": report})
-        return "", chat_history, format_urls(urls)
+            # Sidebar for URLs
+            with gr.Column(scale=1):
+                gr.Markdown("### 🔗 References")
+                url_display = gr.Markdown("(No sources yet)")
 
-    # Chain both: show user → respond with URLs
-    msg.submit(show_user_msg, [msg, chatbot], [chatbot], queue=False).then(
-        respond, [msg, chatbot], [msg, chatbot, url_display]
-    )
+        # Step 1: show user message immediately
+        def show_user_msg(message, chat_history: list):
+            chat_history.append({"role": "user", "content": message})
+            return chat_history
+
+        # Step 2: respond asynchronously (simulate the agent)
+        async def respond(message: str, chat_history: list):
+            report, urls = await interact_with_agent(message)
+            chat_history.append({"role": "assistant", "content": report})
+            return "", chat_history, format_urls(urls)
+
+        # Chain both: show user → respond with URLs
+        # msg.submit(show_user_msg, [msg, chatbot], [chatbot], queue=False).then(
+        #     respond, [msg, chatbot], [msg, chatbot, url_display]
+        # )
+        msg.submit(
+            start_new_query, [msg], [msg, chatbot, url_display], queue=False
+        ).then(
+            lambda message, chat_history: [
+                {"role": "user", "content": message}
+            ],  # user message
+            [msg, chatbot],
+            [chatbot],
+        ).then(respond, [msg, chatbot], [msg, chatbot, url_display])
 
-chat_app.launch()
+    chat_app.launch()