naas-abi 1.0.14__tar.gz → 1.0.15__tar.gz

{naas_abi-1.0.14 → naas_abi-1.0.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi
-Version: 1.0.14
+Version: 1.0.15
 Summary: Multi-agent orchestrator and knowledge graph management system for AI orchestration, providing comprehensive coordination of specialized AI agents and semantic data management capabilities
 Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
 License: MIT License

{naas_abi-1.0.14 → naas_abi-1.0.15}/naas_abi/agents/AbiAgent.py

@@ -1,14 +1,11 @@
 from typing import Optional
 
 from langchain_core.tools import tool
-from naas_abi_core.services.agent.IntentAgent import (
-    AgentConfiguration,
-    AgentSharedState,
-    Intent,
-    IntentAgent,
-    IntentScope,
-    IntentType,
-)
+from naas_abi_core.module.Module import BaseModule
+from naas_abi_core.services.agent.IntentAgent import (AgentConfiguration,
+                                                      AgentSharedState, Intent,
+                                                      IntentAgent, IntentScope,
+                                                      IntentType)
 
 NAME = "Abi"
 AVATAR_URL = (
@@ -149,9 +146,12 @@ You can browse the data and run queries there."""
 
     tools.append(open_knowledge_graph_explorer)
 
-    from naas_abi_core.modules.templatablesparqlquery import (
-        ABIModule as TemplatableSparqlQueryABIModule,
-    )
+    from naas_abi import ABIModule
+    from naas_abi_core.modules.templatablesparqlquery import \
+        ABIModule as TemplatableSparqlQueryABIModuleModule
+    
+    templatable_sparql_query_module : BaseModule = ABIModule.get_instance().engine.modules["naas_abi_core.modules.templatablesparqlquery"]
+    assert isinstance(templatable_sparql_query_module, TemplatableSparqlQueryABIModuleModule), "TemplatableSparqlQueryABIModuleModule must be a subclass of BaseModule"
 
     agent_recommendation_tools = [
         "find_business_proposal_agents",
@@ -161,7 +161,7 @@ You can browse the data and run queries there."""
         "find_fastest_agents",
         "find_cheapest_agents",
     ]
-    sparql_query_tools_list = TemplatableSparqlQueryABIModule.get_instance().get_tools(
+    sparql_query_tools_list = templatable_sparql_query_module.get_tools(
         agent_recommendation_tools
     )
     tools += sparql_query_tools_list
@@ -176,7 +176,6 @@ You can browse the data and run queries there."""
 
     # Define agents - all agents are now loaded automatically during module loading
     agents: list = []
-    from naas_abi import ABIModule
     from naas_abi_core import logger
 
     modules = ABIModule.get_instance().engine.modules.values()

{naas_abi-1.0.14 → naas_abi-1.0.15}/naas_abi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi
-Version: 1.0.14
+Version: 1.0.15
 Summary: Multi-agent orchestrator and knowledge graph management system for AI orchestration, providing comprehensive coordination of specialized AI agents and semantic data management capabilities
 Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
 License: MIT License

{naas_abi-1.0.14 → naas_abi-1.0.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "naas-abi"
-version = "1.0.14"
+version = "1.0.15"
 description = "Multi-agent orchestrator and knowledge graph management system for AI orchestration, providing comprehensive coordination of specialized AI agents and semantic data management capabilities"
 readme = "README.md"
 authors = [{ name = "Maxime Jublou", email = "maxime@naas.ai" },{ name = "Florent Ravenel", email = "florent@naas.ai" }, { name = "Jeremy Ravenel", email = "jeremy@naas.ai" }]