fix: Add correct AsyncGenerator class. Resolve streaming error in stream_agent method

.gitignore

@@ -33,4 +33,6 @@ ENV/
 *.swo
 
 # Project specific
-test_axiom/
+test_axiom/
+.chainlit/
+.files/

chainlit.md

@@ -0,0 +1,14 @@
+# Welcome to Chainlit! 🚀🤖
+
+Hi there, Developer! 👋 We're excited to have you on board. Chainlit is a powerful tool designed to help you prototype, debug and share applications built on top of LLMs.
+
+## Useful Links 🔗
+
+- **Documentation:** Get started with our comprehensive [Chainlit Documentation](https://docs.chainlit.io) 📚
+- **Discord Community:** Join our friendly [Chainlit Discord](https://discord.gg/k73SQ3FyUh) to ask questions, share your projects, and connect with other developers! 💬
+
+We can't wait to see what you create with Chainlit! Happy coding! 💻😊
+
+## Welcome screen
+
+To modify the welcome screen, edit the `chainlit.md` file at the root of your project. If you do not want a welcome screen, just leave this file empty.

src/axiom/agent.py

@@ -13,7 +13,7 @@ from agents.mcp import MCPServer
 from .config import settings
 from .prompts import AXIOM_AGENT_PROMPT
 
-from types import AsyncGeneratorType
+from typing_extensions import AsyncGenerator
 
 class AxiomAgent:
     def __init__(
@@ -29,8 +29,8 @@ class AxiomAgent:
         self.agent = Agent(
             name="Axiom 2.0",
             instructions=AXIOM_AGENT_PROMPT,
-            mcp_servers=mcp_servers,
-            tools=tools,
+            mcp_servers=mcp_servers if mcp_servers is not None else [],
+            tools=tools if tools is not None else [],
         )
 
     def _get_model_config(self):
@@ -59,7 +59,7 @@ class AxiomAgent:
     async def stream_agent(self, input: str | list[dict[str, str]]) -> AsyncGenerator:
         config = self._get_model_config()
 
-        result = await Runner.run_streamed(
+        result = Runner.run_streamed(
             starting_agent=self.agent,
             input=input,
             run_config=config