langchain-dev-utils 1.2.11__py3-none-any.whl → 1.2.12__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.2.11"
+__version__ = "1.2.12"

langchain_dev_utils/agents/middleware/__init__.py

@@ -1,3 +1,4 @@
+from .format_prompt import format_prompt
 from .model_fallback import ModelFallbackMiddleware
 from .model_router import ModelRouterMiddleware
 from .plan import (
@@ -22,4 +23,5 @@ __all__ = [
     "LLMToolEmulator",
     "ModelRouterMiddleware",
     "ToolCallRepairMiddleware",
+    "format_prompt",
 ]

langchain_dev_utils/agents/middleware/format_prompt.py

@@ -0,0 +1,66 @@
+from langchain.agents.middleware import ModelRequest, dynamic_prompt
+from langchain_core.prompts.string import get_template_variables
+
+
+@dynamic_prompt
+def format_prompt(request: ModelRequest) -> str:
+    """Format the system prompt with variables from state and context.
+
+    This middleware function extracts template variables from the system prompt
+    and populates them with values from the agent's state and runtime context.
+    Variables are first resolved from the state, then from the context if not found.
+
+    Example:
+        >>> from langchain_dev_utils.agents.middleware.format_prompt import format_prompt
+        >>> from langchain.agents import create_agent
+        >>> from langchain_core.messages import HumanMessage
+        >>> from dataclasses import dataclass
+        >>>
+        >>> @dataclass
+        ... class Context:
+        ...       name: str
+        ...       user: str
+        >>>
+        >>> agent=create_agent(
+        ...     model=model,
+        ...     tools=tools,
+        ...     system_prompt="You are a helpful assistant. Your name is {name}. Your user is {user}.",
+        ...     middleware=[format_prompt],
+        ...     context_schema=Context,
+        ... )
+        >>> agent.invoke(
+        ...     {
+        ...         "messages": [HumanMessage(content="Hello")],
+        ...     },
+        ...     context=Context(name="assistant", user="tbice"),
+        ... )
+
+    """
+    system_msg = request.system_message
+    if system_msg is None:
+        raise ValueError(
+            "system_message must be provided,while use format_prompt in middleware."
+        )
+
+    system_prompt = "\n".join(
+        [content.get("text", "") for content in system_msg.content_blocks]
+    )
+    variables = get_template_variables(system_prompt, "f-string")
+
+    format_params = {}
+
+    state = request.state
+    for key in variables:
+        if var := state.get(key, None):
+            format_params[key] = var
+
+    other_var_keys = set(variables) - set(format_params.keys())
+
+    if other_var_keys:
+        context = request.runtime.context
+        if context is not None:
+            for key in other_var_keys:
+                if var := getattr(context, key, None):
+                    format_params[key] = var
+
+    return system_prompt.format(**format_params)

{langchain_dev_utils-1.2.11.dist-info → langchain_dev_utils-1.2.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.2.11
+Version: 1.2.12
 Summary: A practical utility library for LangChain and LangGraph development
 Project-URL: Source Code, https://github.com/TBice123123/langchain-dev-utils
 Project-URL: repository, https://github.com/TBice123123/langchain-dev-utils

{langchain_dev_utils-1.2.11.dist-info → langchain_dev_utils-1.2.12.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langchain_dev_utils/__init__.py,sha256=aBWZsCYiXXcSUsUJr3tOTQWsH7ZDqJzyMYdQbOd5Qtc,23
+langchain_dev_utils/__init__.py,sha256=dbW85A2PinQCZabwD2DNDTfOE9315GDtQQKAsJP8IXk,23
 langchain_dev_utils/_utils.py,sha256=MFEzR1BjXMj6HEVwt2x2omttFuDJ_rYAEbNqe99r9pM,1338
 langchain_dev_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_dev_utils/agents/__init__.py,sha256=PJ-lSDZv_AXMYA3H4fx-HzJa14tPbkGmq1HX8LNfaPo,125
@@ -6,7 +6,8 @@ langchain_dev_utils/agents/factory.py,sha256=XdGjktksfTDys7X4SgfPrQz10HUo5fTNAWE
 langchain_dev_utils/agents/file_system.py,sha256=Yk3eetREE26WNrnTWLoiDUpOyCJ-rhjlfFDk6foLa1E,8468
 langchain_dev_utils/agents/plan.py,sha256=WwhoiJBmVYVI9bT8HfjCzTJ_SIp9WFil0gOeznv2omQ,6497
 langchain_dev_utils/agents/wrap.py,sha256=RuchoH_VotPmKFuYEn2SXoSgNxZhSA9jKM0Iv_8oHLk,4718
-langchain_dev_utils/agents/middleware/__init__.py,sha256=sAd0gehREpt0MB5deX5_YmTJPBoqmeSsjNQ-ta6R3EM,768
+langchain_dev_utils/agents/middleware/__init__.py,sha256=EECbcYcHXQAMA-guJNRGwCVi9jG957d0nOaoIuyIKC0,832
+langchain_dev_utils/agents/middleware/format_prompt.py,sha256=rfii98tmOqkjaNHxWy7hovhEYKXrF0CdzsMLO54_CDI,2359
 langchain_dev_utils/agents/middleware/model_fallback.py,sha256=nivtXXF4cwyOBv6p7RW12nXtNg87wjTWxO3BKIYiroI,1674
 langchain_dev_utils/agents/middleware/model_router.py,sha256=pOK-4PNTLrmjaQA9poHoQnsaVwoX0JeJrLVysulv9iU,7631
 langchain_dev_utils/agents/middleware/plan.py,sha256=0qDCmenxgY_zrwMfOyYlgLfhYNw-HszNLeeOkfj14NA,16002
@@ -31,7 +32,7 @@ langchain_dev_utils/pipeline/types.py,sha256=T3aROKKXeWvd0jcH5XkgMDQfEkLfPaiOhhV
 langchain_dev_utils/tool_calling/__init__.py,sha256=mu_WxKMcu6RoTf4vkTPbA1WSBSNc6YIqyBtOQ6iVQj4,322
 langchain_dev_utils/tool_calling/human_in_the_loop.py,sha256=7Z_QO5OZUR6K8nLoIcafc6osnvX2IYNorOJcbx6bVso,9672
 langchain_dev_utils/tool_calling/utils.py,sha256=S4-KXQ8jWmpGTXYZitovF8rxKpaSSUkFruM8LDwvcvE,2765
-langchain_dev_utils-1.2.11.dist-info/METADATA,sha256=4IGI_v0rIC8_SW13_j3h2JyPS_-rKPvdK6m6lTW2ymE,11853
-langchain_dev_utils-1.2.11.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.2.11.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.2.11.dist-info/RECORD,,
+langchain_dev_utils-1.2.12.dist-info/METADATA,sha256=5MjLniCOhGN8a0L6cNLD69PUMNzf5K0nS4Cfifdl5CI,11853
+langchain_dev_utils-1.2.12.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.2.12.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.2.12.dist-info/RECORD,,