ibm-watsonx-orchestrate 1.6.2__py3-none-any.whl → 1.6.4__py3-none-any.whl

ibm_watsonx_orchestrate/flow_builder/utils.py

@@ -6,7 +6,6 @@ import logging
 from pydantic import BaseModel, TypeAdapter
 
 from langchain_core.utils.json_schema import dereference_refs
-from rpds import List
 import typer
 import yaml
 
@@ -87,9 +86,15 @@ def _get_tool_request_body(schema_obj: JsonSchemaObject) -> ToolRequestBody:
         return None
     
     if isinstance(schema_obj, JsonSchemaObject):
-        request_obj = ToolRequestBody(type='object', properties=schema_obj.properties, required=schema_obj.required)
-        if schema_obj.model_extra:
-            request_obj.__pydantic_extra__ = schema_obj.model_extra
+        if schema_obj.type == "object":
+            request_obj = ToolRequestBody(type='object', properties=schema_obj.properties, required=schema_obj.required)
+            if schema_obj.model_extra:
+                request_obj.__pydantic_extra__ = schema_obj.model_extra
+        else:  # we need to wrap a simple type with an object
+            request_obj = ToolRequestBody(type='object', properties={}, required=[])
+            request_obj.properties["data"] = schema_obj
+            if schema_obj.model_extra:
+                request_obj.__pydantic_extra__ = schema_obj.model_extra
 
         return request_obj
     
@@ -162,17 +167,27 @@ async def import_flow_model(model):
 
     return tool_id
 
-def import_flow_support_tools():
+def import_flow_support_tools(model):
 
     if not is_local_dev():
         # we can't import support tools into non-local environments yet
         return []
 
+        
+    schedulable = False
+    if "schedulable" in model["spec"]:
+        schedulable = model["spec"]["schedulable"]
+
     client = instantiate_client(TempusClient)
 
     logger.info(f"Import 'get_flow_status' tool spec...")
     tools = [create_flow_status_tool("i__get_flow_status_intrinsic_tool__")]
 
+    if schedulable:
+        get_schedule_tool = create_get_schedule_tool("i__get_schedule_intrinsic_tool__")
+        delete_schedule_tool = create_delete_schedule_tool("i__delete_schedule_intrinsic_tool__")
+        tools.extend([get_schedule_tool, delete_schedule_tool])
+
     return tools
 
 # Assisted by watsonx Code Assistant
@@ -187,7 +202,7 @@ def create_flow_status_tool(flow_status_tool: str, TEMPUS_ENDPOINT: str="http://
     )
 
     openapi_binding = OpenApiToolBinding(
-        http_path="/flows",
+        http_path="/v1/flows",
         http_method="GET",
         security=[],
         servers=[TEMPUS_ENDPOINT]
@@ -213,3 +228,102 @@ def create_flow_status_tool(flow_status_tool: str, TEMPUS_ENDPOINT: str="http://
 
     return OpenAPITool(spec=spec)
 
+
+def create_get_schedule_tool(name: str, TEMPUS_ENDPOINT: str="http://wxo-tempus-runtime:9044") -> dict:
+
+    spec = ToolSpec(
+        name=name,
+        description="Use this tool to show the current schedules.",
+        permission='read_only',
+        display_name= "Get Schedules"
+    )
+
+    openapi_binding = OpenApiToolBinding(
+        http_path="/v1/schedules/simple",
+        http_method="GET",
+        security=[],
+        servers=[TEMPUS_ENDPOINT]
+    )
+    
+    spec.binding = ToolBinding(openapi=openapi_binding)
+    # Input Schema
+    properties = {
+        "query_schedule_id": {
+            "type": "string",
+            "title": "schedule_id",
+            "description": "Identifies the schedule instance.",
+            "in": "query"
+        },
+        "query_schedule_name": {
+            "type": "string",
+            "title": "schedule_name",
+            "description": "Identifies the schedule name.",
+            "in": "query"
+        },
+    }
+    
+    spec.input_schema = ToolRequestBody(
+        type='object',
+        properties=properties,
+        required=[]
+    )
+
+    response_properties = {
+        "schedule_id": {
+            "type": "string",
+        },
+        "schedule_name": {
+            "type": "string",
+        },
+        "schedule_data": {
+            "type": "string",
+        },
+        "schedule_time": {
+            "type": "string",
+        }
+    }
+
+    spec.output_schema = ToolResponseBody(type='object',
+                                          properties=response_properties,
+                                          description='Return the information about the schedule.')
+
+    return OpenAPITool(spec=spec)
+
+
+def create_delete_schedule_tool(name: str, TEMPUS_ENDPOINT: str="http://wxo-tempus-runtime:9044") -> dict:
+
+    spec = ToolSpec(
+        name=name,
+        description="Use this tool to delete/remove a schedule based on the schedule_id.",
+        permission='read_only',
+        display_name= "Delete Schedule"
+    )
+
+    openapi_binding = OpenApiToolBinding(
+        http_path="/v1/schedules/{schedule_id}",
+        http_method="DELETE",
+        security=[],
+        servers=[TEMPUS_ENDPOINT]
+    )
+    
+    spec.binding = ToolBinding(openapi=openapi_binding)
+    # Input Schema
+    properties = {
+        "path_schedule_id": {
+            "type": "string",
+            "title": "schedule_id",
+            "description": "Identifies the schedule instance.",
+            "in": "query"
+        }
+    }
+    
+    spec.input_schema = ToolRequestBody(
+        type='object',
+        properties=properties,
+        required=[]
+    )
+
+    spec.output_schema = ToolResponseBody(type='object',
+                                          description='Schedule deleted.')
+
+    return OpenAPITool(spec=spec)

ibm_watsonx_orchestrate/utils/exceptions.py

@@ -0,0 +1,23 @@
+import logging
+import sys
+import os
+
+logger = logging.getLogger(__name__)
+
+class BadRequest(Exception):
+    def __init__(self, message: str):
+        super().__init__(message)
+        self.message = message
+        logger.error(message)
+
+        # We need to exit to avoid getting 2 error messages printed
+        # We don't want to exit while running tests
+        # Only exit if not running in a test and no --debug
+        if not self._in_test() and "--debug" not in sys.argv:
+            sys.exit(1)
+
+    def _in_test(self):
+        return "PYTEST_CURRENT_TEST" in os.environ
+
+    def __str__(self):
+        return self.message

{ibm_watsonx_orchestrate-1.6.2.dist-info → ibm_watsonx_orchestrate-1.6.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ibm-watsonx-orchestrate
-Version: 1.6.2
+Version: 1.6.4
 Summary: IBM watsonx.orchestrate SDK
 Author-email: IBM <support@ibm.com>
 License: MIT License
@@ -10,15 +10,11 @@ Requires-Dist: certifi>=2024.8.30
 Requires-Dist: click<8.2.0,>=8.0.0
 Requires-Dist: docstring-parser<1.0,>=0.16
 Requires-Dist: httpx<1.0.0,>=0.28.1
-Requires-Dist: ibm-cloud-sdk-core>=3.22.0
-Requires-Dist: ibm-watsonx-orchestrate-evaluation-framework==1.0.6
+Requires-Dist: ibm-cloud-sdk-core>=3.24.2
+Requires-Dist: ibm-watsonx-orchestrate-evaluation-framework==1.0.8
 Requires-Dist: jsonref==1.1.0
-Requires-Dist: jsonschema<5.0.0,>=4.23.0
-Requires-Dist: langchain-community<=0.3.25,>=0.3.12
 Requires-Dist: langchain-core<=0.3.63
 Requires-Dist: langsmith<=0.3.45
-Requires-Dist: munch>=4.0.0
-Requires-Dist: numpy>=1.26.0
 Requires-Dist: packaging>=24.2
 Requires-Dist: pydantic<3.0.0,>=2.10.3
 Requires-Dist: pyjwt<3.0.0,>=2.10.1

{ibm_watsonx_orchestrate-1.6.2.dist-info → ibm_watsonx_orchestrate-1.6.4.dist-info}/RECORD

@@ -1,19 +1,19 @@
-ibm_watsonx_orchestrate/__init__.py,sha256=OfVb1qB0WmlqIpIZShHMh2YTgizxQfQjlz2yaO6-tyE,425
+ibm_watsonx_orchestrate/__init__.py,sha256=Tv41P2ZTNFM2lUmG__AsJAXcEe8c8riWQvpLyEHlqio,426
 ibm_watsonx_orchestrate/agent_builder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/agent_builder/agents/__init__.py,sha256=lmZwaiWXD4Ea19nrMwZXaqCxFMG29xNS8vUoZtK3yI4,392
-ibm_watsonx_orchestrate/agent_builder/agents/agent.py,sha256=I8ZpOKOGXAmhUIwusDmLqCUawSg74-WZFJ5IijruxAQ,932
-ibm_watsonx_orchestrate/agent_builder/agents/assistant_agent.py,sha256=U1cKr22umcWCMgI4G3r5JZXHPfWYbCbiTydH4YZy_Og,987
-ibm_watsonx_orchestrate/agent_builder/agents/external_agent.py,sha256=LQM-DfOg2ajzrEur_F-aSrqKqaX4TbeajRe9YWz6s7E,981
-ibm_watsonx_orchestrate/agent_builder/agents/types.py,sha256=E5frrJPL7URhH6lD3QsO2heyndsBrfFWDRxXiIdH_nI,12095
+ibm_watsonx_orchestrate/agent_builder/agents/agent.py,sha256=W0uya81fQPrYZFaO_tlsxBL56Bfpw0xrqdxQJhAZ6XI,983
+ibm_watsonx_orchestrate/agent_builder/agents/assistant_agent.py,sha256=NnWThJ2N8HUOD9IDL6ZhtTKyLMHSacJCpxDNityRmgY,1051
+ibm_watsonx_orchestrate/agent_builder/agents/external_agent.py,sha256=7HzEFjd7JiiRTgvA1RVA3M0-Mr42FTQnOtGMII5ufk0,1045
+ibm_watsonx_orchestrate/agent_builder/agents/types.py,sha256=kXbWWz4iHo-knQ5OZRZW7kc8rTwfOzPVn7iyndk5cDs,13013
 ibm_watsonx_orchestrate/agent_builder/agents/webchat_customizations/__init__.py,sha256=5TXa8UqKUAlDo4hTbE5S9OPEkQhxXhPJHJC4pEe8U00,92
 ibm_watsonx_orchestrate/agent_builder/agents/webchat_customizations/prompts.py,sha256=jNVF_jgz1Dmt7-RxAceAS0XWXk_fx9h3sS_fGrvZT28,941
 ibm_watsonx_orchestrate/agent_builder/agents/webchat_customizations/welcome_content.py,sha256=U76wZrblSXx4qv7phcPYs3l8SiFzwZ5cJ74u8Y2iYhU,608
 ibm_watsonx_orchestrate/agent_builder/connections/__init__.py,sha256=VG45ifpp6B00-ZAGW-dIEHFh18Fx3wbyfUiDDKS74u4,718
-ibm_watsonx_orchestrate/agent_builder/connections/connections.py,sha256=UiqTps0yRTSIFoQ0hS6GbpRMlk3PzZmzXEwHlZTTGO8,5359
-ibm_watsonx_orchestrate/agent_builder/connections/types.py,sha256=Lu_LPHKIo_vuptNixU8Uu_j8d_YT85Fjy9rMNYxnXGA,9253
+ibm_watsonx_orchestrate/agent_builder/connections/connections.py,sha256=CD3QHI4d-j3rTugHOgvPZUdRu_zpWCwsSzkN63_Ke_c,5423
+ibm_watsonx_orchestrate/agent_builder/connections/types.py,sha256=YOOlXXNKVxTZ6tK2ceeytWXUApZjwfi4mgdYwXoiqOk,9532
 ibm_watsonx_orchestrate/agent_builder/knowledge_bases/knowledge_base.py,sha256=_KuGF0RZpKpwdt31rzjlTjrhGRFz2RtLzleNkhMNX4k,1831
-ibm_watsonx_orchestrate/agent_builder/knowledge_bases/knowledge_base_requests.py,sha256=mRow0rf5EHeJCCsTrONeTq10jShs_yIBQjpgDY_mTrI,1641
-ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py,sha256=ttre2pRs_I_KBWSht60zT3dJw8twQml3m92ZkiRAoxs,7994
+ibm_watsonx_orchestrate/agent_builder/knowledge_bases/knowledge_base_requests.py,sha256=3xTfFMZR17EN8eYRhsVyBfOEzlTqyi0eYaMXyv0_ZtQ,862
+ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py,sha256=_jHu9ELQqj1qwDK6U7jqemzNFeM2O5APlykiZRsDOcY,7009
 ibm_watsonx_orchestrate/agent_builder/model_policies/__init__.py,sha256=alJEjlneWlGpadmvOVlDjq5wulytKOmpkq3849fhKNc,131
 ibm_watsonx_orchestrate/agent_builder/model_policies/types.py,sha256=a6f9HP2OlZIe36k_PDRmFtefz2Ms2KBpzJ_jz8ggYbk,882
 ibm_watsonx_orchestrate/agent_builder/models/__init__.py,sha256=R5nTbyMBzahONdp5-bJFp-rbtTDnp2184k6doZqt67w,31
@@ -21,40 +21,42 @@ ibm_watsonx_orchestrate/agent_builder/models/types.py,sha256=Zr0xHZpTVPZnaUSvlop
 ibm_watsonx_orchestrate/agent_builder/toolkits/base_toolkit.py,sha256=KXRPgBK-F9Qa6IYqEslkN3ANj3cmZoZQnlSiy_-iXCk,1054
 ibm_watsonx_orchestrate/agent_builder/toolkits/types.py,sha256=yY-V4Hqct91-Rs4rJ3rY9OhzKkSMdOT63o224o-U9eg,959
 ibm_watsonx_orchestrate/agent_builder/tools/__init__.py,sha256=adkYX0wgB-RKFCUBw6LPJhNVelUjUdsxipGPk2ghLns,479
-ibm_watsonx_orchestrate/agent_builder/tools/base_tool.py,sha256=unJBOJUY8DAq3T3YX5d1H5KehJUCjObAdpGLVoWIfzw,1156
+ibm_watsonx_orchestrate/agent_builder/tools/base_tool.py,sha256=0vwMIAyKyB8v1QmNrubLy8Al58g3qT78EUgrmOjegoI,1220
 ibm_watsonx_orchestrate/agent_builder/tools/flow_tool.py,sha256=DJWYVmIjw1O_cbzPpwU0a_vIZGvo0mj8UsjW9zkKMlA,2589
-ibm_watsonx_orchestrate/agent_builder/tools/openapi_tool.py,sha256=Q2-TcUUjBAZWlhOqxfkB3yQpdxgdZAAJh0VWDh_dZPQ,16359
-ibm_watsonx_orchestrate/agent_builder/tools/python_tool.py,sha256=qVyAe7hKLZ1V4cvqDUCpYs6JBKl9MHSr0yrri-nSceA,10600
-ibm_watsonx_orchestrate/agent_builder/tools/types.py,sha256=TXxWCSddXqRL2GbjkxT4vYJcRMjWS7g22wMMrBz-Wj8,7934
+ibm_watsonx_orchestrate/agent_builder/tools/openapi_tool.py,sha256=h9ma18GUQHt88UIakd6QZHic822bAXzYokh6sfqAMZk,19410
+ibm_watsonx_orchestrate/agent_builder/tools/python_tool.py,sha256=FDY-y9jy4bNy88i7JG85Zj_YrxnPfJPD0DjwvL9ITxc,12012
+ibm_watsonx_orchestrate/agent_builder/tools/types.py,sha256=SAyDEnWYuguTFk6WxgSDnMaONqDqcr6QhBAxD1u5FDU,8165
 ibm_watsonx_orchestrate/agent_builder/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ibm_watsonx_orchestrate/agent_builder/utils/pydantic_utils.py,sha256=QEanM6FpkmntvS02whdhWx1d4v6zT_1l9ipEbfTgHs8,7623
 ibm_watsonx_orchestrate/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ibm_watsonx_orchestrate/cli/config.py,sha256=BE2ze0o9vLoyrk7q9mcRek2rDL2cTsVtsSGWw4W0y6M,8212
-ibm_watsonx_orchestrate/cli/init_helper.py,sha256=Lb0bj6b8EaeBxkfGyrBBvc581ORg_S5-tHcXWQ-mBmE,1466
-ibm_watsonx_orchestrate/cli/main.py,sha256=TyVuNC12GDxo4InaPfQ6e6Xk5bGarfSoIRAslGifUQ0,2914
+ibm_watsonx_orchestrate/cli/config.py,sha256=iXjDxymWhAmLSUu6eh7zJR20dYZDzbxcU5VoBdh3VIw,8318
+ibm_watsonx_orchestrate/cli/init_helper.py,sha256=qxnKdFcPtGsV_6RqP_IuLshRxgB004SxzDAkBTExA-4,1675
+ibm_watsonx_orchestrate/cli/main.py,sha256=eE71dfx-EdDYHymYDqXcTjyMXpX0b3cueXOxCI2YLLs,3119
 ibm_watsonx_orchestrate/cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ibm_watsonx_orchestrate/cli/commands/agents/agents_command.py,sha256=-UX6ypiF_XwmEwe0twVGsfnO29Xs8xPOPJ5vk8EIifo,8991
-ibm_watsonx_orchestrate/cli/commands/agents/agents_controller.py,sha256=zTk3AemWOldfgKYYR1MOcT7z2JbYdgZgSsi7KxugFGw,47934
+ibm_watsonx_orchestrate/cli/commands/agents/agents_command.py,sha256=Zif3bH3X7j5qf7MWGcr53mJ2oEKXNRHeg3ZHeoHTIqQ,8984
+ibm_watsonx_orchestrate/cli/commands/agents/agents_controller.py,sha256=AvDZ3HSRFzaCQ00Kgu9IsOtq0Dv6HPv2E0o9IrisA7Y,47981
 ibm_watsonx_orchestrate/cli/commands/channels/channels_command.py,sha256=fVIFhPUTPdxsxIE10nWL-W5wvBR-BS8V8D6r__J8R98,822
 ibm_watsonx_orchestrate/cli/commands/channels/channels_controller.py,sha256=WjQxwJujvo28SsWgfJSXIpkcgniKcskJ2arL4MOz0Ys,455
-ibm_watsonx_orchestrate/cli/commands/channels/types.py,sha256=h8gA3EM7k5X08sk8T9b5siL-XL9uUE-hl_jS-0PDBHA,254
+ibm_watsonx_orchestrate/cli/commands/channels/types.py,sha256=hMFvWPr7tAmDrhBqtzfkCsrubX3lsU6lapTSOFsUbHM,475
 ibm_watsonx_orchestrate/cli/commands/channels/webchat/channels_webchat_command.py,sha256=vPCr6z1n1yzGDjTlM4f3_9MiuVRzNmXbvUifm6XyQi8,1103
-ibm_watsonx_orchestrate/cli/commands/channels/webchat/channels_webchat_controller.py,sha256=0S1ch8SwkGRtAyy7nAZEjKl92hFkIE9TU71JJ82zdjU,8128
+ibm_watsonx_orchestrate/cli/commands/channels/webchat/channels_webchat_controller.py,sha256=CGfmKsCBX4E3HMZ8C0IXD-DHQNwe96V1Y_BxUZM2us0,8557
 ibm_watsonx_orchestrate/cli/commands/chat/chat_command.py,sha256=Q9vg2Z5Fsunu6GQFY_TIsNRhUCa0SSGSPnK4jxSGK34,1581
-ibm_watsonx_orchestrate/cli/commands/connections/connections_command.py,sha256=qgg2NbAB7g_PHPHd0ZLUm0EjyaCSPi1qpOhtyOlydcY,10151
-ibm_watsonx_orchestrate/cli/commands/connections/connections_controller.py,sha256=ka7uBpuRlICZ4ACGiT4WqmufZYEw566NlZX5FhB9pXM,22178
+ibm_watsonx_orchestrate/cli/commands/connections/connections_command.py,sha256=bZBzaaVMFmGY4Guk-JCknZqd8HvXY5L-FirxpxddQfc,10497
+ibm_watsonx_orchestrate/cli/commands/connections/connections_controller.py,sha256=odsqxthEDyPO3C-GEtSn31sxLjRYgUBf1Vdkg6ZEwms,22290
+ibm_watsonx_orchestrate/cli/commands/copilot/copilot_command.py,sha256=IxasApIyQYWRMKPXKa38ZPVkUvOc4chggSmSGjgQGXc,2345
+ibm_watsonx_orchestrate/cli/commands/copilot/copilot_controller.py,sha256=rr6p4_yqriq3ts1FwFC4aV0ub5UC5D6JqpuLoba4j_U,14808
+ibm_watsonx_orchestrate/cli/commands/copilot/copilot_server_controller.py,sha256=AcBE97qNYsRN0ftY_E-AAjKFyVea4fHtU5eB2HsB42I,5619
 ibm_watsonx_orchestrate/cli/commands/environment/environment_command.py,sha256=xwq7gdyjMtl2RYAnLAahGk2wDetr9BStt8yu7JkeBhk,3768
-ibm_watsonx_orchestrate/cli/commands/environment/environment_controller.py,sha256=8f4yH-hmPJFo-wAHA61XjRh_gst7JIIiRBQDFj0dXHs,10183
-ibm_watsonx_orchestrate/cli/commands/environment/types.py,sha256=qKcgIUfTPJIomLor-utz4qi7YhBMpb1l2DiSvjVSgaA,175
-ibm_watsonx_orchestrate/cli/commands/evaluations/evaluations_command.py,sha256=YHaTxcQxLkuojFhNcTtYbzb_lTv_dYBuz2IHl9RXXwU,11689
-ibm_watsonx_orchestrate/cli/commands/evaluations/evaluations_controller.py,sha256=pTWyuPNHKOPCYTMlYWwoGojZBdvDrHme2EAel0l4ltk,6957
-ibm_watsonx_orchestrate/cli/commands/knowledge_bases/knowledge_bases_command.py,sha256=Hxj33rFEYo1Vr2bk-uRwRhNssAETtTQgWTREvC0Nxds,3348
-ibm_watsonx_orchestrate/cli/commands/knowledge_bases/knowledge_bases_controller.py,sha256=pbsjkBssLOTSMfh5ymgDAq9DxVdD01eAI33si0JZqS4,9631
+ibm_watsonx_orchestrate/cli/commands/environment/environment_controller.py,sha256=oHZ7LONtPg3-SSnU_rRZryLi8N2mplz5h-LGg4XjzD4,10261
+ibm_watsonx_orchestrate/cli/commands/environment/types.py,sha256=X6jEnyBdxakromA7FhQ5btZMj9kwGcwRSFz8vpD65jA,224
+ibm_watsonx_orchestrate/cli/commands/evaluations/evaluations_command.py,sha256=nOVxeZSTp1bfV_l_06B6x6wfNeusNAr5KImJYkwGWx8,14298
+ibm_watsonx_orchestrate/cli/commands/evaluations/evaluations_controller.py,sha256=dZEAD0rIS9DQjWD2-i6367RjNd2PWB3Fm_DDk25toBg,7855
+ibm_watsonx_orchestrate/cli/commands/knowledge_bases/knowledge_bases_command.py,sha256=hOzRcGVoqq7dTc4bSregKxH-kYbrVqaFdhBLawqnRNo,2668
+ibm_watsonx_orchestrate/cli/commands/knowledge_bases/knowledge_bases_controller.py,sha256=M19rRMFEEcAarPVw4fOr3S94Se1ZEo4hD7sCUr0xYTI,10115
 ibm_watsonx_orchestrate/cli/commands/login/login_command.py,sha256=xArMiojoozg7Exn6HTpbTcjDO2idZRA-y0WV-_Ic1Sk,651
 ibm_watsonx_orchestrate/cli/commands/models/model_provider_mapper.py,sha256=RVZfFCuPWiASkR9Mox61v5SZ1AoRxo4_g9stnbfjCkc,7994
-ibm_watsonx_orchestrate/cli/commands/models/models_command.py,sha256=xYbMHRM98TCBmnwgIOsqpfjWqD7HVirn6A4KyEgQEcE,6418
-ibm_watsonx_orchestrate/cli/commands/models/models_controller.py,sha256=BNfUoWST60x3k7yIvSbtBR3rmyTQmz9g0TDK2WOoOE4,18610
-ibm_watsonx_orchestrate/cli/commands/server/server_command.py,sha256=VstqSn3dzuzziwEc0VVkf1y9E3qN94mTJuCbeFO4KCY,37478
+ibm_watsonx_orchestrate/cli/commands/models/models_command.py,sha256=PW-PIM5Nq0qdCopWjANGBWEuEoA3NLTFThYrN8ggGCI,6425
+ibm_watsonx_orchestrate/cli/commands/models/models_controller.py,sha256=eZSYQUg9TL_-8lgcPVpKIx7MtOE7K_NCvZW9Y9YsFA0,18466
+ibm_watsonx_orchestrate/cli/commands/server/server_command.py,sha256=ho1rSTN_fIJKeKzA5d4BdoQELl_4aF5Gzx90VySbaX8,39687
 ibm_watsonx_orchestrate/cli/commands/server/types.py,sha256=UCrgGErbSVBnOzesfjrIii4tTCuVfWemYz5AKGZX0oA,4213
 ibm_watsonx_orchestrate/cli/commands/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/cli/commands/settings/settings_command.py,sha256=CzXRkd-97jXyS6LtaaNtMah-aZu0919dYl-mDwzGThc,344
@@ -62,20 +64,20 @@ ibm_watsonx_orchestrate/cli/commands/settings/observability/__init__.py,sha256=4
 ibm_watsonx_orchestrate/cli/commands/settings/observability/observability_command.py,sha256=TAkpKwoqocsShSgEeR6LzHCzJx16VDQ6cYsbpljxeqI,372
 ibm_watsonx_orchestrate/cli/commands/settings/observability/langfuse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/cli/commands/settings/observability/langfuse/langfuse_command.py,sha256=Wa0L8E44EdxH9LdOvmnluLk_ApJVfTLauNOC1kV4W8k,6515
-ibm_watsonx_orchestrate/cli/commands/toolkit/toolkit_command.py,sha256=e8YKUr_8CodA5fgYjUKj-Dk856foQ5Iz3I9Jld5pU_o,4193
+ibm_watsonx_orchestrate/cli/commands/toolkit/toolkit_command.py,sha256=e2VwpxbN7vwfFp6Gu8hDe2B5c3fdzXYfQBkN15tYR90,4195
 ibm_watsonx_orchestrate/cli/commands/toolkit/toolkit_controller.py,sha256=fq48tWNbuYFjrHXyVDmJBT96tww5iMdE2iOYUjr-fUc,11563
 ibm_watsonx_orchestrate/cli/commands/tools/tools_command.py,sha256=Cuo1ZvlfsymojqbadCqdwwS0HUjaWpe2XQrV70g61_s,3943
-ibm_watsonx_orchestrate/cli/commands/tools/tools_controller.py,sha256=z8Q_t6ABrzJ2nnB7YYK7MCzDCWL9Jt2wfkfdDDJ-Pe4,39480
+ibm_watsonx_orchestrate/cli/commands/tools/tools_controller.py,sha256=uL0WHi0YD7NIYSLOui-CymVHj-h9Y0i3PxS7T2SXSvo,40408
 ibm_watsonx_orchestrate/cli/commands/tools/types.py,sha256=_md0GEa_cTH17NO_moWDY_LNdFvyEFQ1UVB9_FltYiA,173
 ibm_watsonx_orchestrate/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ibm_watsonx_orchestrate/client/base_api_client.py,sha256=NklphjtCpSFroBocOp4OiklC0iIF0JzXJjtJhUd8gZ8,4752
+ibm_watsonx_orchestrate/client/base_api_client.py,sha256=cSbQb7-K9KoMeFLEIiYGho4cQqf8TcPyYklju9X4cC4,5214
 ibm_watsonx_orchestrate/client/base_service_instance.py,sha256=sM_r7bln9BpgEOhaJMdFI9-je3T7GLQxLduk-in0oRY,235
 ibm_watsonx_orchestrate/client/client.py,sha256=pgYeXHe4_Makrw0gTyM343DQdrZZB8cjitFHKcbdzuc,2432
 ibm_watsonx_orchestrate/client/client_errors.py,sha256=72MKCNZbKoo2QXyY0RicLhP3r0ALRjgOEbHOHNSyOYI,11712
 ibm_watsonx_orchestrate/client/credentials.py,sha256=gDVeeQZDdbbjJiO1EI61yx2oRgTQctxA2ZesSDHI4DA,3786
-ibm_watsonx_orchestrate/client/local_service_instance.py,sha256=gdeol7xNRVnxEMr5rui1ovqtwDfMRziHn9GIjo7-MVo,3447
-ibm_watsonx_orchestrate/client/service_instance.py,sha256=uZlsMW9LC_2GggqEjBQY09fm6uZJ-0audwzqgCyjbbQ,5277
-ibm_watsonx_orchestrate/client/utils.py,sha256=86UH5H54uTJA2kFb4dXrnGfs7hFQZ3h21ElwngZpT24,4745
+ibm_watsonx_orchestrate/client/local_service_instance.py,sha256=dt7vfLnjgt7mT8wSq8SJZndNTwsPzhb0XDhcnPUPFpU,3524
+ibm_watsonx_orchestrate/client/service_instance.py,sha256=fp3Lc4yQf4zTkxVS5WnIAkrHT0xG_a5i44qcLeQkaa4,6600
+ibm_watsonx_orchestrate/client/utils.py,sha256=MUw11r0_wYv3RdF6-1BmSxcwYNF6q2_Urte7tMYTKVA,5836
 ibm_watsonx_orchestrate/client/agents/agent_client.py,sha256=v1ZqXKj3p5JeJyOLpV84UUI2LY8XPjW1ejo5yzkeiMQ,4708
 ibm_watsonx_orchestrate/client/agents/assistant_agent_client.py,sha256=1JQN0E4T_uz5V0LM-LD1ahNu2KCeFBjXAr8WCiP9mkE,1745
 ibm_watsonx_orchestrate/client/agents/external_agent_client.py,sha256=iQ44XBdC4rYfS-zFn4St1xC5y5gf5SNqKHzMNQcFDZc,1808
@@ -83,9 +85,10 @@ ibm_watsonx_orchestrate/client/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5
 ibm_watsonx_orchestrate/client/analytics/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/client/analytics/llm/analytics_llm_client.py,sha256=0YS_BCpmf5oGFawpZkJ38cuz5ArhKsZIbSydWRd194s,1340
 ibm_watsonx_orchestrate/client/connections/__init__.py,sha256=J7TOyVg38h71AlaJjlFs5fOuAXTceHvELtOJ9oz4Mvg,207
-ibm_watsonx_orchestrate/client/connections/connections_client.py,sha256=B6LQPwoFHg_CIXHEJBjVu6fXnRj_xVsLGsC-DhmdfSQ,8478
+ibm_watsonx_orchestrate/client/connections/connections_client.py,sha256=8f_olWVgQVLi6xKO_DvV68zu_AMLe1DMhoQAhC-BG3g,7583
 ibm_watsonx_orchestrate/client/connections/utils.py,sha256=f6HsiDI6cycOqfYN6P4uZ3SQds83xlh83zTUioZPeYk,2618
-ibm_watsonx_orchestrate/client/knowledge_bases/knowledge_base_client.py,sha256=U-pG_H0I8992f0V13Li_e1dksKp54MrYX3X9bvr-09w,2181
+ibm_watsonx_orchestrate/client/copilot/cpe/copilot_cpe_client.py,sha256=eGE5VJb47XZxiMszBVQ7u0NKTVeHPuGVyfRJ6qPKlVs,1939
+ibm_watsonx_orchestrate/client/knowledge_bases/knowledge_base_client.py,sha256=YjC16dcL_l0jd-6yZ6uH8anlL4yaIv9f-7Y_urPuXWM,2201
 ibm_watsonx_orchestrate/client/model_policies/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/client/model_policies/model_policies_client.py,sha256=Ddjraesv1MRPhZebB0PdBL0zgdsoWmnYpWTUci_6XFI,2258
 ibm_watsonx_orchestrate/client/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -93,8 +96,8 @@ ibm_watsonx_orchestrate/client/models/models_client.py,sha256=ZvP3iPgUFw_SXp41kJ
 ibm_watsonx_orchestrate/client/toolkit/toolkit_client.py,sha256=TLFNS39EeBD_t4Y-rX9sGp4sWBDr--mE5qVtHq8Q2hk,3121
 ibm_watsonx_orchestrate/client/tools/tempus_client.py,sha256=24fKDZUOVHBW-Vj4mubnpnUmab5LgGn8u5hOVyJaozI,1804
 ibm_watsonx_orchestrate/client/tools/tool_client.py,sha256=d3i3alVwa0TCN72w9sWOrM20GCbNmnpTnqEOJVbBIFM,1994
-ibm_watsonx_orchestrate/docker/compose-lite.yml,sha256=TvddzCD2nFAGbu-1Hf2AAhdTORoTsll2ssa1lJVsoZE,32856
-ibm_watsonx_orchestrate/docker/default.env,sha256=5JnLinuQoN3HxaApodke2SO-rI_pDtC2WecpL0s7pek,5329
+ibm_watsonx_orchestrate/docker/compose-lite.yml,sha256=igtukwqVFamdaRbuAbrlLPU8EfZ7hpveWSA0Ka34YHo,39437
+ibm_watsonx_orchestrate/docker/default.env,sha256=84p3BEFeA0mVe_C_WKMi4JLiOhQkFLq8hwI_Hoe1eWE,5894
 ibm_watsonx_orchestrate/docker/proxy-config-single.yaml,sha256=WEbK4ENFuTCYhzRu_QblWp1_GMARgZnx5vReQafkIG8,308
 ibm_watsonx_orchestrate/docker/start-up.sh,sha256=LTtwHp0AidVgjohis2LXGvZnkFQStOiUAxgGABOyeUI,1811
 ibm_watsonx_orchestrate/docker/sdk/ibm_watsonx_orchestrate-0.6.0-py3-none-any.whl,sha256=Hi3-owh5OM0Jz2ihX9nLoojnr7Ky1TV-GelyqLcewLE,2047417
@@ -102,24 +105,24 @@ ibm_watsonx_orchestrate/docker/sdk/ibm_watsonx_orchestrate-0.6.0.tar.gz,sha256=e
 ibm_watsonx_orchestrate/docker/tempus/common-config.yaml,sha256=Zo3F36F5DV4VO_vUg1RG-r4WhcukVh79J2fXhGl6j0A,22
 ibm_watsonx_orchestrate/flow_builder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/flow_builder/data_map.py,sha256=1brmWWFERDsNG2XGais-5-r58LKUUwBtqwdaLQIFRhE,503
-ibm_watsonx_orchestrate/flow_builder/node.py,sha256=tp_ssBOSDDi8q-ET2bP7xRPwNOLmLUgQ7v9zZZqn9SU,4090
-ibm_watsonx_orchestrate/flow_builder/types.py,sha256=KPASSlw55Dzp_8nhFdJUN0UI8_knMzLCK9epVcfP5NY,27689
-ibm_watsonx_orchestrate/flow_builder/utils.py,sha256=gKIKtb6hOfImt4cll4LOGgDoEUHubcz-AFq1byCpmaM,8203
-ibm_watsonx_orchestrate/flow_builder/flows/__init__.py,sha256=UTwpiAlT9Wytn1kK-kgUaY6UYjLKsiaQK0ZdSbKLtM0,897
+ibm_watsonx_orchestrate/flow_builder/node.py,sha256=I90a-K_xnEgZrl2NHo_rDmfT2Dj0n0Om15aDnEzhe74,4755
+ibm_watsonx_orchestrate/flow_builder/types.py,sha256=bpnodCo9PnacSCvKHEybx8F8rvcp55DXITrXUmt2q88,37142
+ibm_watsonx_orchestrate/flow_builder/utils.py,sha256=1ZetA6GBCG5MGqhGWDM66R_VuyRZ1pYCtVeeIRQgNIg,11709
+ibm_watsonx_orchestrate/flow_builder/flows/__init__.py,sha256=dHS2fB-xbVS5yt_yNyZqYeuiAGs78lbCF65AX_VQlD4,1025
 ibm_watsonx_orchestrate/flow_builder/flows/constants.py,sha256=XXYtL5Y1OTjj3jKo2zoJiwb7FCUNu_M43fgU8w591TY,322
-ibm_watsonx_orchestrate/flow_builder/flows/decorators.py,sha256=k9-r9ly5ih5oHBqjEgrc2vfeVC4LOMKg3YVLacBDubM,2702
-ibm_watsonx_orchestrate/flow_builder/flows/events.py,sha256=bhbN3ljl-WyiJ_LzMd99IlX0QrhB_bASyHWv8mtTjvo,2978
-ibm_watsonx_orchestrate/flow_builder/flows/flow.py,sha256=1Tc8L8rGbocPVdUNk-Zp_Fi9QxGLSg22LWPewlH7GKU,51020
-ibm_watsonx_orchestrate/flow_builder/resources/flow_status.openapi.yml,sha256=UkQ4FD_ZhvZuMOjrgLm7cx8zJFZD7Ri-MPCe_zktU-8,1664
+ibm_watsonx_orchestrate/flow_builder/flows/decorators.py,sha256=lr4qSWq5PWqlGFf4fzUQZCVQDHBYflrYwZ24S89Aq80,2794
+ibm_watsonx_orchestrate/flow_builder/flows/events.py,sha256=VyaBm0sADwr15LWfKbcBQS1M80NKqzYDj3UlW3OpOf4,2984
+ibm_watsonx_orchestrate/flow_builder/flows/flow.py,sha256=a33ciPFM5VUQlY4BdcXAQVed_MGARNi_GpChIHJ1uS4,55124
 ibm_watsonx_orchestrate/run/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/run/connections.py,sha256=K-65GXPA8GEsVmRdPfMe_LB2G9RfXQUr95wvRUOhkS4,1828
 ibm_watsonx_orchestrate/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ibm_watsonx_orchestrate/utils/exceptions.py,sha256=MjpoGuk7lMziPT2Zo7X-jgGlm0M22mTZF4Y8EHb3KPM,631
 ibm_watsonx_orchestrate/utils/utils.py,sha256=U7z_2iASoFiZ2zM0a_2Mc2Y-P5oOT7hOwuurzC9Z3N8,721
 ibm_watsonx_orchestrate/utils/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/utils/logging/logger.py,sha256=FzeGnidXAjC7yHrvIaj4KZPeaBBSCniZFlwgr5yV3oA,1037
 ibm_watsonx_orchestrate/utils/logging/logging.yaml,sha256=9_TKfuFr1barnOKP0fZT5D6MhddiwsXVTFjtRbcOO5w,314
-ibm_watsonx_orchestrate-1.6.2.dist-info/METADATA,sha256=vEKjA3XqPvwQinfXD0axdqgnN-QQlYGeSvXOmXjc27Q,1510
-ibm_watsonx_orchestrate-1.6.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ibm_watsonx_orchestrate-1.6.2.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
-ibm_watsonx_orchestrate-1.6.2.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
-ibm_watsonx_orchestrate-1.6.2.dist-info/RECORD,,
+ibm_watsonx_orchestrate-1.6.4.dist-info/METADATA,sha256=TTYaKeV208aYXY8edBO897I7H93VRBtsRa4WO2m-D3U,1360
+ibm_watsonx_orchestrate-1.6.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ibm_watsonx_orchestrate-1.6.4.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
+ibm_watsonx_orchestrate-1.6.4.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
+ibm_watsonx_orchestrate-1.6.4.dist-info/RECORD,,

ibm_watsonx_orchestrate/agent_builder/utils/pydantic_utils.py

@@ -1,149 +0,0 @@
-import json
-from pathlib import Path
-from typing import Any, Dict, Type, Literal
-
-import jsonschema
-from pydantic import BaseModel
-from pydantic.json_schema import DEFAULT_REF_TEMPLATE, GenerateJsonSchema, JsonSchemaMode
-from pydantic.main import IncEx
-from typing_extensions import Self
-
-from ibm_watsonx_orchestrate.agent_builder.tools import ToolRequestBody, ToolResponseBody, JsonSchemaObject
-
-
-def generate_schema_only_base_model(schema: ToolRequestBody | ToolResponseBody | JsonSchemaObject) -> Type[BaseModel]:
-    class SchemaOnlyBaseModel(BaseModel):
-        __primitive__: Any
-        model_config = {
-            'extra': 'allow'
-        }
-        """
-        The purpose of a SchemaOnlyBaseModel is to pass along the json schema represented by schema
-        to a langchain tool's arg_schema
-        :arg schema The json schema to emulate
-        :returns a fake BaseModel that only returns a json schema
-        """
-
-        def __init__(self, /, *args, **kwargs: Any) -> None:
-            if schema.type == 'object':
-                super().__init__(**kwargs)
-                for name, value in kwargs.items():
-                    setattr(self, name, value)
-            else:
-                kwargs={}
-                kwargs['__primitive__'] = args[0]
-                super().__init__(**kwargs)
-                setattr(self, '__primitive__', args[0])
-
-        @classmethod
-        def model_json_schema(cls, by_alias: bool = True, ref_template: str = DEFAULT_REF_TEMPLATE,
-                              schema_generator: type[GenerateJsonSchema] = GenerateJsonSchema,
-                              mode: JsonSchemaMode = 'validation') -> dict[str, Any]:
-            return schema.model_dump(exclude_unset=True, exclude_none=True)
-
-        @classmethod
-        def schema(cls, by_alias: bool = True, ref_template: str = DEFAULT_REF_TEMPLATE) -> Dict[str, Any]:
-            return schema.model_dump(exclude_unset=True, exclude_none=True)
-
-        @classmethod
-        def schema_json(cls, *, by_alias: bool = True, ref_template: str = DEFAULT_REF_TEMPLATE,
-                        **dumps_kwargs: Any) -> str:
-            return schema.model_dump_json(exclude_unset=True, exclude_none=True)
-
-        @classmethod
-        def model_validate_json(cls, json_data: str | bytes | bytearray, *, strict: bool | None = None,
-                                context: Any | None = None) -> Self:
-            obj = json.loads(json_data)
-            jsonschema.validate(obj, schema=schema.model_dump(exclude_unset=True))
-            if schema.type == 'object':
-                return SchemaOnlyBaseModel(**obj)
-            else:
-                return SchemaOnlyBaseModel(obj)
-
-        @classmethod
-        def model_validate(cls, obj: Any, *, strict: bool | None = None, from_attributes: bool | None = None,
-                           context: Any | None = None) -> Self:
-            jsonschema.validate(obj, schema=schema.model_dump(exclude_unset=True))
-            if schema.type == 'object':
-                return SchemaOnlyBaseModel(**obj)
-            else:
-                return SchemaOnlyBaseModel(obj)
-
-        @classmethod
-        def model_validate_strings(cls, obj: Any, *, strict: bool | None = None, context: Any | None = None) -> Self:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def validate(cls, value: Any) -> Self:
-            jsonschema.validate(value, schema=schema.model_dump(exclude_unset=True))
-            return SchemaOnlyBaseModel(**value)
-
-        @classmethod
-        def model_construct(cls, _fields_set: set[str] | None = None, **values: Any) -> Self:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def model_parametrized_name(cls, params: tuple[type[Any], ...]) -> str:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def model_rebuild(cls, *, force: bool = False, raise_errors: bool = True, _parent_namespace_depth: int = 2,
-                          _types_namespace: Any = None) -> bool | None:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def parse_obj(cls, obj: Any) -> Self:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def parse_raw(cls, b: str | bytes, *, content_type: str | None = None, encoding: str = 'utf8',
-                      proto: Any | None = None, allow_pickle: bool = False) -> Self:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def parse_file(cls, path: str | Path, *, content_type: str | None = None, encoding: str = 'utf8',
-                       proto: Any | None = None, allow_pickle: bool = False) -> Self:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def from_orm(cls, obj: Any) -> Self:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def construct(cls, _fields_set: set[str] | None = None, **values: Any) -> Self:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        @classmethod
-        def update_forward_refs(cls, **localns: Any) -> None:
-            raise NotImplementedError('unimplemented for SchemaOnlyBaseModel')
-
-        def model_dump(self, *, mode: Literal['json', 'python'] | str = 'python', include: IncEx | None = None,
-                       exclude: IncEx | None = None, context: Any | None = None, by_alias: bool = False,
-                       exclude_unset: bool = False, exclude_defaults: bool = False, exclude_none: bool = False,
-                       round_trip: bool = False, warnings: bool | Literal['none', 'warn', 'error'] = True,
-                       serialize_as_any: bool = False) -> dict[str, Any]:
-            primitive = getattr(self, '__primitive__', None)
-            if primitive is not None:
-                return primitive
-            else:
-                return super().model_dump(mode=mode, include=include, exclude=exclude, context=context, by_alias=by_alias,
-                                          exclude_unset=exclude_unset, exclude_defaults=exclude_defaults,
-                                          exclude_none=exclude_none, round_trip=round_trip, warnings=warnings,
-                                          serialize_as_any=serialize_as_any)
-
-        def model_dump_json(self, *, indent: int | None = None, include: IncEx | None = None,
-                            exclude: IncEx | None = None, context: Any | None = None, by_alias: bool = False,
-                            exclude_unset: bool = False, exclude_defaults: bool = False, exclude_none: bool = False,
-                            round_trip: bool = False, warnings: bool | Literal['none', 'warn', 'error'] = True,
-                            serialize_as_any: bool = False) -> str:
-            primitive = getattr(self, '__primitive__')
-            if primitive is not None:
-                return json.dumps(primitive)
-            else:
-                return super().model_dump_json(indent=indent, include=include, exclude=exclude, context=context,
-                                           by_alias=by_alias, exclude_unset=exclude_unset,
-                                           exclude_defaults=exclude_defaults, exclude_none=exclude_none,
-                                           round_trip=round_trip, warnings=warnings, serialize_as_any=serialize_as_any)
-
-    return SchemaOnlyBaseModel
-

ibm_watsonx_orchestrate/flow_builder/resources/flow_status.openapi.yml

@@ -1,66 +0,0 @@
-openapi: 3.0.3
-info:
-  title: Flow status
-  version: '0.1'
-  description: This tool allows you to check the status of a flow.
-servers:
-  - url: http://wxo-tempus-runtime:9044
-components:
-  schemas:
-    APIError:
-      type: object
-      properties:
-        data:
-          type: object
-          properties:
-            message:
-              type: string
-          additionalProperties: true
-          required:
-            - message
-      required:
-        - data
-paths:
-  /v1/flows:
-    get:
-      description: Get the status of a flow instance.
-      operationId: get_flow_status
-      responses:
-        '200':
-          description: Return the status of a flow instance.
-          content:
-            application/json:
-              schema:
-                type: array
-                items:
-                  type: object
-        '400':
-          description: Bad input
-          content:
-            application/json:
-              schema:
-                $ref: '#/components/schemas/APIError'
-        '500':
-          description: Internal server error
-          content:
-            application/json:
-              schema:
-                $ref: '#/components/schemas/APIError'
-      parameters:
-        - in: query
-          name: instance_id
-          description: Identifies the instance of the flow. This was provided when the flow tool was run.
-          required: false
-          schema:
-            type: string
-        - in: query
-          name: state
-          required: false
-          schema:
-            type: string
-            enum:
-              - completed
-              - in_progress
-              - interrupted
-              - failed
-