async functions need prompt

src/upgrade_advisor/agents/package.py

@@ -57,7 +57,7 @@ class PackageDiscoveryAgent:
         self.agent = CodeAgent(
             tools=tool_list,
             model=model,
-            max_steps=10,
+            max_steps=20,
             add_base_tools=True,
             additional_authorized_imports=[
                 "json",
@@ -73,7 +73,8 @@ class PackageDiscoveryAgent:
                 "markdownify",
                 "sys",
                 "tomli",
-                "requests",
+                # "requests",
+                "asyncio",
             ],
         )
         logger.info(

src/upgrade_advisor/agents/tools/pypi_api.py

@@ -31,11 +31,11 @@ async def pypi_search(
         dict: Parsed package metadata or an error payload.
     """
     REQUEST_URL = f"https://pypi.python.org/pypi/{package}/json"
+
     response = requests.get(REQUEST_URL, timeout=10)
     if response.ok:
         result = parse_response_pypi_search(response.json(), cutoff=cutoff)
         return result.model_dump()
-
     e = HTTPError(str(response.status_code))
     return ErrorResponseSchema(error=str(e)).model_dump()
 

src/upgrade_advisor/agents/tools/tools.py

@@ -186,7 +186,9 @@ class PypiSearchTool(Tool):
     name = "pypi_search"
     description = """
         Get metadata about a PyPI package by its name.
-        It returns a dictionary with the schema described in `output_schema` attribute.
+        It returns a dictionary with the schema described in `output_schema`
+        attribute.
+        This is an async tool, so make sure to await its execution.
         """
     inputs = {
         "package": {
@@ -204,8 +206,8 @@ class PypiSearchTool(Tool):
     def __init__(self):
         super().__init__()
 
-    def forward(self, package: str, cutoff: int) -> dict:
-        result = pypi_search(package, cutoff=cutoff)
+    async def forward(self, package: str, cutoff: int) -> dict:
+        result = await pypi_search(package, cutoff=cutoff)
         return result
 
 
@@ -215,7 +217,9 @@ class PypiSearchVersionTool(Tool):
     name = "pypi_search_version"
     description = """
         Get metadata about a specific version of a PyPI package. 
-        It returns a dictionary with the schema described in `output_schema` attribute.
+        It returns a dictionary with the schema described in `output_schema`
+        attribute.
+        This is an async tool, so make sure to await its execution.
         """
     inputs = {
         "package": {
@@ -237,8 +241,8 @@ class PypiSearchVersionTool(Tool):
     def __init__(self):
         super().__init__()
 
-    def forward(self, package: str, version: str, cutoff: int) -> dict:
-        result = pypi_search_version(package, version, cutoff=cutoff)
+    async def forward(self, package: str, version: str, cutoff: int) -> dict:
+        result = await pypi_search_version(package, version, cutoff=cutoff)
         return result
 
 
@@ -280,7 +284,9 @@ class RepoFromPyPITool(Tool):
         PyPI.
         Some projects may not have a GitHub repository listed in their PyPI
         metadata.
-        It returns a dictionary with the schema described in `output_schema` attribute.
+        It returns a dictionary with the schema described in `output_schema`
+        attribute.
+        This is an async tool, so make sure to await its execution.
         """
     inputs = {
         "package": {
@@ -298,7 +304,7 @@ class RepoFromPyPITool(Tool):
     def __init__(self):
         super().__init__()
 
-    def forward(self, package: str, cutoff: int) -> dict:
-        result = github_repo_and_releases(package, cutoff=cutoff)
+    async def forward(self, package: str, cutoff: int) -> dict:
+        result = await github_repo_and_releases(package, cutoff=cutoff)
 
         return result

src/upgrade_advisor/theme.py

@@ -40,14 +40,14 @@ class Christmas(Base):
         )
         super().set(
             body_background_fill=(
-                "radial-gradient(#ffffff 2px, transparent 2px) 0 0/26px 26px, "
-                "radial-gradient(#ef4444 2px, transparent 2px) 13px 13px/26px 26px, "
-                "#0f2a1d"
+                "radial-gradient(rgba(255, 255, 255, 0.6) 2px, transparent 2px) 0 0/26px 26px, "
+                "radial-gradient(rgba(239, 68, 68, 0.55) 2px, transparent 2px) 13px 13px/26px 26px, "
+                "rgba(15, 42, 29, 0.92)"
             ),
             body_background_fill_dark=(
-                "radial-gradient(#f3f4f6 2px, transparent 2px) 0 0/26px 26px, "
-                "radial-gradient(#f87171 2px, transparent 2px) 13px 13px/26px 26px, "
-                "#05140d"
+                "radial-gradient(rgba(243, 244, 246, 0.52) 2px, transparent 2px) 0 0/26px 26px, "
+                "radial-gradient(rgba(248, 113, 113, 0.48) 2px, transparent 2px) 13px 13px/26px 26px, "
+                "rgba(5, 20, 13, 0.94)"
             ),
             background_fill_primary="rgba(255, 255, 255, 0.92)",
             background_fill_primary_dark="rgba(12, 22, 17, 0.9)",
@@ -71,9 +71,9 @@ class Christmas(Base):
             block_title_text_color_dark="*primary_100",
             block_title_text_weight="700",
             button_primary_background_fill="linear-gradient(120deg, *primary_500, #f59e0b 80%)",
-            button_primary_background_fill_hover="linear-gradient(120deg, *primary_400, #fbbf24 80%)",
+            button_primary_background_fill_hover="linear-gradient(120deg, #34d399, #bbf7d0 80%)",
             button_primary_background_fill_dark="linear-gradient(120deg, *primary_600, #d97706 80%)",
-            button_primary_background_fill_hover_dark="linear-gradient(120deg, *primary_500, #f59e0b 80%)",
+            button_primary_background_fill_hover_dark="linear-gradient(120deg, #10b981, #34d399 80%)",
             button_primary_text_color="white",
             button_primary_text_color_dark="white",
             button_primary_shadow="0 10px 40px rgba(0, 0, 0, 0.25)",