aap-core 0.1.2.dev1__tar.gz → 0.1.3.dev1__tar.gz

{aap_core-0.1.2.dev1 → aap_core-0.1.3.dev1}/PKG-INFO

@@ -1,7 +1,8 @@
 Metadata-Version: 2.3
 Name: aap-core
-Version: 0.1.2.dev1
+Version: 0.1.3.dev1
 Summary: Agent design pattern core framework
+Keywords: agent,ai,pattern,llm
 Author: Ly Hon Quang
 Author-email: Ly Hon Quang <lyhonquang@gmail.com>
 License: MIT
@@ -21,3 +22,4 @@ Requires-Python: >=3.10.12
 Project-URL: Repository, https://github.com/quanghona/agent_design_pattern.git
 Description-Content-Type: text/markdown
 
+# AI Agent Pattern (AAP)

aap_core-0.1.3.dev1/README.md

@@ -0,0 +1 @@
+# AI Agent Pattern (AAP)

{aap_core-0.1.2.dev1 → aap_core-0.1.3.dev1}/pyproject.toml

@@ -1,9 +1,10 @@
 [project]
 name = "aap_core"
-version = "0.1.2.dev1"
+version = "0.1.3.dev1"
 description = "Agent design pattern core framework"
 readme = "README.md"
 license = { text = "MIT" }
+keywords = ["agent", "ai", "pattern", "llm"]
 authors = [
     { name = "Ly Hon Quang", email = "lyhonquang@gmail.com" }
 ]

{aap_core-0.1.2.dev1 → aap_core-0.1.3.dev1}/src/aap_core/chain.py

@@ -5,9 +5,9 @@ from typing import Callable, Generic, List, Tuple
 from pydantic import Field, PrivateAttr
 
 from .guardrail import BaseGuardRail, PassGuardRail
-from .prompt_enhancer import (
-    BasePromptEnhancer,
-    IdentityPromptEnhancer,
+from .prompt_augmenter import (
+    BasePromptAugmenter,
+    IdentityPromptAugmenter,
 )
 from .types import AgentMessage, BaseChain, ChainMessage, ChainResponse
 
@@ -108,8 +108,8 @@ class TypicalLLMChain(BaseLLMChain):
         default=PassGuardRail(),
         description="validate the input message",
     )
-    prompt_enhancer: BasePromptEnhancer = Field(
-        default=IdentityPromptEnhancer(),
+    prompt_augmenter: BasePromptAugmenter = Field(
+        default=IdentityPromptAugmenter(),
         description="add more context to the prompt or rewrite / refine the prompt",
     )
     tools: Sequence[Callable] = Field(
@@ -128,21 +128,17 @@ class TypicalLLMChain(BaseLLMChain):
 
     def invoke(self, message: AgentMessage, **kwargs) -> AgentMessage:
         message = self.input_guardrail(message)
-        message = self.prompt_enhancer(message)
+        message = self.prompt_augmenter(message)
         message = self.generate(message, **kwargs)
         message = self.output_guardrail(message)
         return message
 
 
-class LLMPromptEnhancer(BasePromptEnhancer):
-    """A prompt enhancer that use an LLM chain to rewrite the prompt."""
+class MetaPromptAugmenter(BasePromptAugmenter):
+    """A prompt augmenter that use an LLM chain to rewrite the prompt."""
 
     chain: BaseLLMChain = Field(..., description="LLM chain that rewrite the prompt")
 
-    def __call__(self, message: AgentMessage, **kwargs) -> AgentMessage:
+    def augment(self, message: AgentMessage, **kwargs) -> AgentMessage:
         message = self.chain.invoke(message, **kwargs)
-        if message.execution_result != "success":
-            return message
-        _, message.query = message.responses[-1]
-        message.responses = []
         return message

{aap_core-0.1.2.dev1 → aap_core-0.1.3.dev1}/src/aap_core/orchestration.py

@@ -334,6 +334,7 @@ class CoordinatorAgent(BaseAgent):
         # TODO: check step dependency and parallelize steps to speed up
         summary_steps_key = self.summary_steps_key.replace("context_", "")
         for i, (step_msg, step_agent, dependencies) in enumerate(steps):
+            # TODO: deal with the dependencies
             step_msg.context = {summary_steps_key: sub_task_result}
             self.state = f"step {i}: worker {step_agent.card.name} running"
             sub_task_result.append(step_agent.execute(step_msg, **kwargs))