naas-abi 1.0.13__py3-none-any.whl → 1.0.14__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- naas_abi/agents/EntitytoSPARQLAgent.py +4 -4
- {naas_abi-1.0.13.dist-info → naas_abi-1.0.14.dist-info}/METADATA +1 -1
- {naas_abi-1.0.13.dist-info → naas_abi-1.0.14.dist-info}/RECORD +5 -5
- {naas_abi-1.0.13.dist-info → naas_abi-1.0.14.dist-info}/WHEEL +1 -1
- {naas_abi-1.0.13.dist-info → naas_abi-1.0.14.dist-info}/top_level.txt +0 -0
|
@@ -8,12 +8,12 @@ from langchain_core.tools import BaseTool, Tool
|
|
|
8
8
|
from langgraph.checkpoint.base import BaseCheckpointSaver
|
|
9
9
|
from langgraph.checkpoint.memory import MemorySaver
|
|
10
10
|
from langgraph.graph import START, StateGraph
|
|
11
|
-
from langgraph.graph.message import MessagesState
|
|
12
11
|
from langgraph.types import Command
|
|
13
12
|
from naas_abi import ABIModule
|
|
14
13
|
from naas_abi_core.engine.EngineProxy import ServicesProxy
|
|
15
14
|
from naas_abi_core.models.Model import ChatModel
|
|
16
15
|
from naas_abi_core.services.agent.Agent import (
|
|
16
|
+
ABIAgentState,
|
|
17
17
|
Agent,
|
|
18
18
|
AgentConfiguration,
|
|
19
19
|
AgentSharedState,
|
|
@@ -168,7 +168,7 @@ def create_agent(
|
|
|
168
168
|
)
|
|
169
169
|
|
|
170
170
|
|
|
171
|
-
class EntityExtractionState(
|
|
171
|
+
class EntityExtractionState(ABIAgentState):
|
|
172
172
|
"""State class for entity extraction conversations.
|
|
173
173
|
|
|
174
174
|
Extends MessagesState to include entity extraction information that tracks
|
|
@@ -930,9 +930,9 @@ INSERT DATA {
|
|
|
930
930
|
This node is used to call the model to extract the entities from the last message.
|
|
931
931
|
"""
|
|
932
932
|
messages = state["messages"]
|
|
933
|
-
if
|
|
933
|
+
if state["system_prompt"]:
|
|
934
934
|
messages = [
|
|
935
|
-
SystemMessage(content=
|
|
935
|
+
SystemMessage(content=state["system_prompt"]),
|
|
936
936
|
] + messages
|
|
937
937
|
|
|
938
938
|
response: BaseMessage = self._chat_model_with_tools.invoke(messages)
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: naas-abi
|
|
3
|
-
Version: 1.0.
|
|
3
|
+
Version: 1.0.14
|
|
4
4
|
Summary: Multi-agent orchestrator and knowledge graph management system for AI orchestration, providing comprehensive coordination of specialized AI agents and semantic data management capabilities
|
|
5
5
|
Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
|
|
6
6
|
License: MIT License
|
|
@@ -2,7 +2,7 @@ naas_abi/__init__.py,sha256=at1v9W4RbH6G94w0tXXLSTeBnI2D_4nrmpAATd-TSdQ,4858
|
|
|
2
2
|
naas_abi/cli.py,sha256=bfCcxGoQ_iSMhQDDX0-dRCyhc53Nawsy3WAruCNCL7M,27549
|
|
3
3
|
naas_abi/agents/AbiAgent.py,sha256=4-o7lebG0_6dLX1rLJ9orXthl7ADQVnenZlgdAR4qRo,18323
|
|
4
4
|
naas_abi/agents/AbiAgent_test.py,sha256=AKggG5RS3zDWnql1DFE7cF7riAHeOdKTZp44WN65Arg,5906
|
|
5
|
-
naas_abi/agents/EntitytoSPARQLAgent.py,sha256=
|
|
5
|
+
naas_abi/agents/EntitytoSPARQLAgent.py,sha256=Njq9su9w7zcpRiKHqX_11jHy7PQUzPl9HmrNmqYiTQ8,57148
|
|
6
6
|
naas_abi/agents/EntitytoSPARQLAgent_test.py,sha256=kx0gkbJoQZqt2_f_RINajwQU5EnHR1PsQx53rh7wJvA,2267
|
|
7
7
|
naas_abi/agents/KnowledgeGraphBuilderAgent.py,sha256=ve1NXFnzU1DNmmq7VD7N9BviYy8-sSnMuEhGzTKe_qY,13008
|
|
8
8
|
naas_abi/agents/KnowledgeGraphBuilderAgent_test.py,sha256=pKgakywjYCkjEngbaG-D9q9gwNMQEgXeD1C2xb9irTo,2718
|
|
@@ -43,7 +43,7 @@ naas_abi/workflows/GetSubjectGraphWorkflow.py,sha256=EkgrFQPyaUVeC8sd7oVcxT9LUka
|
|
|
43
43
|
naas_abi/workflows/GetSubjectGraphWorkflow_test.py,sha256=5B2r3QsdBIKwYOuHkEhnfEPKV6xpWZzwRQyhBintgpo,2067
|
|
44
44
|
naas_abi/workflows/SearchIndividualWorkflow.py,sha256=6vWwywKNBlaosopPxiw0OpQnrn99wuiji0uA7jwKOn4,6572
|
|
45
45
|
naas_abi/workflows/SearchIndividualWorkflow_test.py,sha256=ySEgDQFEbevAxkjX9W8No-7JG9A6bH6Y8bCi86aenVA,2762
|
|
46
|
-
naas_abi-1.0.
|
|
47
|
-
naas_abi-1.0.
|
|
48
|
-
naas_abi-1.0.
|
|
49
|
-
naas_abi-1.0.
|
|
46
|
+
naas_abi-1.0.14.dist-info/METADATA,sha256=7Ahe_jHbJmrXgcHI57_wkzwTjzoNcc9gl-pvedijkjA,15812
|
|
47
|
+
naas_abi-1.0.14.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
|
|
48
|
+
naas_abi-1.0.14.dist-info/top_level.txt,sha256=Dhf3mhATB9-JuvBR8Y55OabvI-PCeQUqb5ae3EMmbeA,9
|
|
49
|
+
naas_abi-1.0.14.dist-info/RECORD,,
|
|
File without changes
|