agno 2.4.7__py3-none-any.whl → 2.4.8__py3-none-any.whl

agno/models/neosantara/neosantara.py

@@ -0,0 +1,42 @@
+from dataclasses import dataclass
+from os import getenv
+from typing import Any, Dict, Optional
+
+from agno.exceptions import ModelAuthenticationError
+from agno.models.openai.like import OpenAILike
+
+
+@dataclass
+class Neosantara(OpenAILike):
+    """
+    A class for interacting with Neosantara API.
+
+    Attributes:
+        id (str): The id of the Neosantara model to use. Default is "grok-4.1-fast-non-reasoning".
+        name (str): The name of this chat model instance. Default is "Neosantara"
+        provider (str): The provider of the model. Default is "Neosantara".
+        api_key (str): The api key to authorize request to Neosantara.
+        base_url (str): The base url to which the requests are sent. Defaults to "https://api.neosantara.xyz/v1".
+    """
+
+    id: str = "grok-4.1-fast-non-reasoning"
+    name: str = "Neosantara"
+    provider: str = "Neosantara"
+    api_key: Optional[str] = None
+    base_url: str = "https://api.neosantara.xyz/v1"
+
+    def _get_client_params(self) -> Dict[str, Any]:
+        """
+        Returns client parameters for API requests, checking for NEOSANTARA_API_KEY.
+
+        Returns:
+            Dict[str, Any]: A dictionary of client parameters for API requests.
+        """
+        if not self.api_key:
+            self.api_key = getenv("NEOSANTARA_API_KEY")
+            if not self.api_key:
+                raise ModelAuthenticationError(
+                    message="NEOSANTARA_API_KEY not set. Please set the NEOSANTARA_API_KEY environment variable.",
+                    model_name=self.name,
+                )
+        return super()._get_client_params()

agno/models/utils.py

@@ -149,6 +149,11 @@ def _get_model_class(model_id: str, model_provider: str) -> Model:
 
         return Nebius(id=model_id)
 
+    elif model_provider == "neosantara":
+        from agno.models.neosantara import Neosantara
+
+        return Neosantara(id=model_id)
+
     elif model_provider == "nexus":
         from agno.models.nexus import Nexus
 

agno/os/app.py

@@ -1056,12 +1056,15 @@ class AgentOS:
         dbs_with_specific_config = [db.db_id for db in knowledge_config.dbs]
 
         # Only add databases that are actually used for knowledge contents
-        for db_id in self.knowledge_dbs.keys():
+        for db_id, dbs in self.knowledge_dbs.items():
             if db_id not in dbs_with_specific_config:
+                # Collect unique table names from all databases with the same id
+                unique_tables = list(set(db.knowledge_table_name for db in dbs))
                 knowledge_config.dbs.append(
                     DatabaseConfig(
                         db_id=db_id,
                         domain_config=KnowledgeDomainConfig(display_name=db_id),
+                        tables=unique_tables,
                     )
                 )
 

agno/os/interfaces/agui/router.py

@@ -14,7 +14,7 @@ try:
     )
     from ag_ui.encoder import EventEncoder
 except ImportError as e:
-    raise ImportError("`ag_ui` not installed. Please install it with `pip install -U ag-ui`") from e
+    raise ImportError("`ag_ui` not installed. Please install it with `pip install -U ag-ui-protocol`") from e
 
 from fastapi import APIRouter
 from fastapi.responses import StreamingResponse

agno/os/routers/components/components.py

@@ -243,6 +243,8 @@ def attach_routes(
                 update_kwargs["description"] = body.description
             if body.metadata is not None:
                 update_kwargs["metadata"] = body.metadata
+            if body.current_version is not None:
+                update_kwargs["current_version"] = body.current_version
             if body.component_type is not None:
                 update_kwargs["component_type"] = DbComponentType(body.component_type)
 

agno/os/routers/knowledge/knowledge.py

@@ -215,7 +215,6 @@ def attach_routes(router: APIRouter, knowledge_instances: List[Union[Knowledge,
         description=(
             "Upload content from a remote source (S3, GCS, SharePoint, GitHub) to the knowledge base. "
             "Content is processed asynchronously in the background. "
-            "Use the /knowledge/config endpoint to see available remote content sources."
         ),
         responses={
             202: {