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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi
-Version: 1.0.13
+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.13 → 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.13 → naas_abi-1.0.15}/naas_abi/agents/EntitytoSPARQLAgent.py

@@ -8,12 +8,12 @@ from langchain_core.tools import BaseTool, Tool
 from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph import START, StateGraph
-from langgraph.graph.message import MessagesState
 from langgraph.types import Command
 from naas_abi import ABIModule
 from naas_abi_core.engine.EngineProxy import ServicesProxy
 from naas_abi_core.models.Model import ChatModel
 from naas_abi_core.services.agent.Agent import (
+    ABIAgentState,
     Agent,
     AgentConfiguration,
     AgentSharedState,
@@ -168,7 +168,7 @@ def create_agent(
     )
 
 
-class EntityExtractionState(MessagesState):
+class EntityExtractionState(ABIAgentState):
     """State class for entity extraction conversations.
 
     Extends MessagesState to include entity extraction information that tracks
@@ -930,9 +930,9 @@ INSERT DATA {
         This node is used to call the model to extract the entities from the last message.
         """
         messages = state["messages"]
-        if self._system_prompt:
+        if state["system_prompt"]:
             messages = [
-                SystemMessage(content=self._system_prompt),
+                SystemMessage(content=state["system_prompt"]),
             ] + messages
 
         response: BaseMessage = self._chat_model_with_tools.invoke(messages)

{naas_abi-1.0.13 → 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.13
+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.13 → naas_abi-1.0.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "naas-abi"
-version = "1.0.13"
+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" }]