xgae 0.2.2__tar.gz → 0.3.0__tar.gz

{xgae-0.2.2 → xgae-0.3.0}/CHANGELOG.md

@@ -1,33 +1,43 @@
-# Release Changelog
-## [0.2.2] - 2025-9-18
+## [0.3.0] - 2025-10-24
 ### Added
-- 
+- Support GAIA2 ARE Engine
+
+
+
+## [0.2.4] - 2025-9-23
+### Modified
+- Project structure
+
+
+## [0.2.3] - 2025-9-19
 ### Modified
 - CustomPromptRag: remove FastEmbedEmbeddings, use 'text-embedding-v3' model for chinese, avoid download 'bge-small-zh-v1.5'
 
+
 ## [0.2.1] - 2025-9-17
 ### Added
-- Example Langgraph ReactAgent: add CustomPromptRag, use FastEmbedEmbeddings and 'BAAI/bge-small-zh-v1.5' model
+- Example ReflectionAgent: add CustomPromptRag, use FastEmbedEmbeddings and 'BAAI/bge-small-zh-v1.5' model
 ### Modified
 - pyproject.toml: add [project.optional-dependencies] 'examples'
 
+
 ## [0.2.0] - 2025-9-10
 ### Added
 - Agent Engine release 0.2
-- Example: Langgraph ReactAgent release 0.2
+- Example: Langgraph ReflectionAgent release 0.2
 ### Fixed
 - Agent Engine: call mcp tool fail, call 'ask' tool again and again
-- Example Langgraph ReactAgent: retry on 'ask', user_input is ask answer
+- Example Langgraph ReflectionAgent: retry on 'ask', user_input is ask answer
 
 
 ## [0.1.20] - 2025-9-9
 ### Added
-- Example: Langgraph ReactAgent add final_result_agent 
+- Example: Langgraph ReflectionAgent add final_result_agent 
 
 
 ## [0.1.19] - 2025-9-8
 ### Added
-- Example: Langgraph ReactAgent release V1, full logic but no final result agent and tool select agent
+- Example: Langgraph ReflectionAgent release V1, full logic but no final result agent and tool select agent
 
 
 # Release Changelog

{xgae-0.2.2 → xgae-0.3.0}/PKG-INFO

@@ -1,11 +1,10 @@
 Metadata-Version: 2.4
 Name: xgae
-Version: 0.2.2
+Version: 0.3.0
 Summary: Extreme General Agent Engine
-Requires-Python: >=3.13
+Requires-Python: >=3.11
 Requires-Dist: colorlog==6.9.0
 Requires-Dist: langchain-mcp-adapters==0.1.9
-Requires-Dist: langchain==0.3.27
 Requires-Dist: langfuse==2.60.9
 Requires-Dist: litellm==1.74.15
 Requires-Dist: mcp==1.13.0

{xgae-0.2.2/src/examples/agent/langgraph/react → xgae-0.3.0/examples/agent/langgraph/reflection}/agent_base.py

@@ -1,4 +1,5 @@
 from typing import Any, Dict, List, TypedDict, Optional
+
 from xgae.engine.engine_base import XGATaskResult
 
 class EvaluateResult(TypedDict, total=False):

{xgae-0.2.2/src/examples/agent/langgraph/react → xgae-0.3.0/examples/agent/langgraph/reflection}/custom_prompt_rag.py

@@ -7,7 +7,7 @@ from langchain_core.embeddings import Embeddings
 from langchain_core.vectorstores import VectorStore
 
 from langchain_community.vectorstores import Chroma
-from chromadb.utils import embedding_functions
+from chromadb.utils.embedding_functions import OpenAIEmbeddingFunction
 
 class ChromaEmbedding(Embeddings):
     embedding_model_name = "text-embedding-v3"
@@ -16,7 +16,7 @@ class ChromaEmbedding(Embeddings):
         api_key = os.getenv('LLM_API_KEY')
         api_base = os.getenv('LLM_API_BASE', "https://dashscope.aliyuncs.com/compatible-mode/v1")
 
-        self.embedding_function = embedding_functions.OpenAIEmbeddingFunction(
+        self.embedding_function = OpenAIEmbeddingFunction(
             api_key = api_key,
             api_base = api_base,
             model_name = self.embedding_model_name,

xgae-0.2.2/src/examples/agent/langgraph/react/react_agent.py → xgae-0.3.0/examples/agent/langgraph/reflection/reflection_agent.py

@@ -7,7 +7,6 @@ from uuid import uuid4
 from langfuse.callback import CallbackHandler
 from langfuse import Langfuse
 
-
 from langgraph.graph import END, START, StateGraph
 from langgraph.types import interrupt, Command
 from langgraph.checkpoint.memory import MemorySaver
@@ -20,11 +19,12 @@ from xgae.engine.engine_base import XGATaskResult
 from xgae.engine.mcp_tool_box import XGAMcpToolBox
 from xgae.engine.task_engine import XGATaskEngine
 
-from examples.agent.langgraph.react.agent_base import AgentContext, TaskState, EvaluateResult
-from examples.agent.langgraph.react.result_eval_agent import TaskResultEvalAgent
-from examples.agent.langgraph.react.custom_prompt_rag import CustomPromptRag
+from examples.agent.langgraph.reflection.agent_base import AgentContext, TaskState, EvaluateResult
+from examples.agent.langgraph.reflection.result_eval_agent import TaskResultEvalAgent
+from examples.agent.langgraph.reflection.custom_prompt_rag import CustomPromptRag
+
 
-class XGAReactAgent:
+class ReflectiontAgent:
     MAX_TASK_RETRY = 2
     QUALIFIED_RESULT_SCORE = 0.7
 
@@ -89,18 +89,24 @@ class XGAReactAgent:
 
 
     def _search_system_prompt(self, user_input: str) -> str:
+        system_prompt = None
         if hasattr(self, 'custom_prompt_rag'):
-            system_prompt = self.custom_prompt_rag.search_prompt(user_input)
+            system_prompt_path = self.custom_prompt_rag.search_prompt(user_input)
+            if system_prompt_path:
+                system_prompt = read_file(system_prompt_path)
         else:
-            system_prompt = None if "fault" not in user_input else read_file("templates/example/fault_user_prompt.txt")
+            if "fault" in user_input:  # only for example
+                system_prompt = read_file("templates/example/fault_user_prompt.txt")
         return system_prompt
 
 
     async def _supervisor_node(self, state: TaskState) -> Dict[str, Any]:
         user_input = state['user_inputs'][0]
         eval_result = state.get('eval_result', None)
+        system_prompt = state.get('system_prompt', None)
 
-        system_prompt = self._search_system_prompt(user_input)
+        if system_prompt is  None and eval_result is None:
+            system_prompt = self._search_system_prompt(user_input)
         is_system_prompt = True if system_prompt is not None else False
 
         general_tools = [] if system_prompt else ["*"]

xgae-0.2.2/src/examples/agent/langgraph/react/run_react_agent.py → xgae-0.3.0/examples/agent/langgraph/reflection/run_agent_app.py

@@ -4,7 +4,7 @@ from uuid import uuid4
 
 from xgae.utils.setup_env import setup_logging
 
-from examples.agent.langgraph.react.react_agent import XGAReactAgent, AgentContext
+from examples.agent.langgraph.reflection.reflection_agent import ReflectiontAgent, AgentContext
 
 
 async def main():
@@ -17,7 +17,7 @@ async def main():
     ]
 
     for user_input in user_inputs:
-        agent = XGAReactAgent(use_prompt_rag=True)
+        agent = ReflectiontAgent(use_prompt_rag=True)
         task_no += 1
         context: AgentContext = {
             'task_id': f"agent_task_{uuid4()}",   # can be set with request_id, must be unique

{xgae-0.2.2 → xgae-0.3.0}/pyproject.toml

@@ -1,15 +1,14 @@
 [project]
 name = "xgae"
-version = "0.2.2"
+version = "0.3.0"
 description = "Extreme General Agent Engine"
 readme = "README.md"
-requires-python = ">=3.13"
+requires-python = ">=3.11"
 dependencies = [
     "colorlog==6.9.0",
     "litellm==1.74.15",
     "mcp==1.13.0",
     "langfuse==2.60.9",
-    "langchain==0.3.27",
     "langchain-mcp-adapters==0.1.9",
 ]
 

xgae-0.3.0/test/test_chroma.py

@@ -0,0 +1,31 @@
+import chromadb
+
+chroma_client = chromadb.EphemeralClient()
+
+# default use 'ONNXMiniLM_L6_V2' embedding function
+collection = chroma_client.get_or_create_collection(name="fault_collection")
+
+collection.upsert(
+    documents=[
+        "Fault location and analysis of fault causes",
+        "This is a analyse about alarm"
+    ],
+    ids=["id1", "id2"],
+    metadatas=[{"type": 1}, {"type": 2}]
+)
+
+querys = ["locate fault and solution",
+          "alarm search"
+          ]
+
+for query in querys:
+    results = collection.query(
+        query_texts=query,
+        n_results=2,
+        #where={"type": 1},
+        where={"type": {"$gt": 0}},
+        where_document = {"$contains": "fault"}
+    )
+    print("*"*50)
+    print(f"query='{query}', id={results['ids'][0]}, score={results['distances'][0]}")
+