Merge pull request #370 from UiPath/akshaya/mockable_update_2

feat(ExampleCalls): add example calls for mocking purposes

Author: akshaylive

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.1.42"
+version = "0.1.43"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath_langchain/agent/tools/context_tool.py

@@ -40,6 +40,7 @@ class ContextOutputSchemaModel(BaseModel):
         description=resource.description,
         input_schema=input_model.model_json_schema(),
         output_schema=output_model.model_json_schema(),
+        example_calls=[],  # Examples cannot be provided for context.
     )
     async def context_tool_fn(query: str) -> dict[str, Any]:
         return {"documents": await retriever.ainvoke(query)}

src/uipath_langchain/agent/tools/escalation_tool.py

@@ -48,6 +48,7 @@ def create_escalation_tool(resource: AgentEscalationResourceConfig) -> Structure
         description=resource.description,
         input_schema=input_model.model_json_schema(),
         output_schema=output_model.model_json_schema(),
+        example_calls=channel.properties.example_calls,
     )
     async def escalation_tool_fn(
         runtime: ToolRuntime, **kwargs: Any

src/uipath_langchain/agent/tools/integration_tool.py

@@ -154,6 +154,7 @@ def create_integration_tool(
         description=resource.description,
         input_schema=input_model.model_json_schema(),
         output_schema=output_model.model_json_schema(),
+        example_calls=resource.properties.example_calls,
     )
     async def integration_tool_fn(**kwargs: Any):
         try:

src/uipath_langchain/agent/tools/process_tool.py

@@ -27,6 +27,7 @@ def create_process_tool(resource: AgentProcessToolResourceConfig) -> StructuredT
         description=resource.description,
         input_schema=input_model.model_json_schema(),
         output_schema=output_model.model_json_schema(),
+        example_calls=resource.properties.example_calls,
     )
     async def process_tool_fn(**kwargs: Any):
         return interrupt(