Merge pull request #57 from langchain-ai/nc/16apr/ii

Small changes to spec

Author: nfcampos

client-python/.openapi-generator/FILES

@@ -17,7 +17,7 @@ ap_client/exceptions.py
 ap_client/models/__init__.py
 ap_client/models/agent.py
 ap_client/models/agent_capabilities.py
-ap_client/models/agent_schemas.py
+ap_client/models/agent_schema.py
 ap_client/models/config.py
 ap_client/models/content.py
 ap_client/models/content_one_of_inner.py
@@ -31,6 +31,7 @@ ap_client/models/run.py
 ap_client/models/run_create.py
 ap_client/models/run_search_request.py
 ap_client/models/run_status.py
+ap_client/models/run_stream.py
 ap_client/models/run_wait_response.py
 ap_client/models/search_agents_request.py
 ap_client/models/search_items_response.py
@@ -50,7 +51,7 @@ ap_client/py.typed
 ap_client/rest.py
 docs/Agent.md
 docs/AgentCapabilities.md
-docs/AgentSchemas.md
+docs/AgentSchema.md
 docs/AgentsApi.md
 docs/BackgroundRunsApi.md
 docs/Config.md
@@ -66,6 +67,7 @@ docs/Run.md
 docs/RunCreate.md
 docs/RunSearchRequest.md
 docs/RunStatus.md
+docs/RunStream.md
 docs/RunWaitResponse.md
 docs/RunsApi.md
 docs/SearchAgentsRequest.md
@@ -91,4 +93,6 @@ setup.cfg
 setup.py
 test-requirements.txt
 test/__init__.py
+test/test_agent_schema.py
+test/test_run_stream.py
 tox.ini

client-python/README.md

@@ -115,7 +115,7 @@ Class | Method | HTTP request | Description
 
  - [Agent](docs/Agent.md)
  - [AgentCapabilities](docs/AgentCapabilities.md)
- - [AgentSchemas](docs/AgentSchemas.md)
+ - [AgentSchema](docs/AgentSchema.md)
  - [Config](docs/Config.md)
  - [Content](docs/Content.md)
  - [ContentOneOfInner](docs/ContentOneOfInner.md)
@@ -129,6 +129,7 @@ Class | Method | HTTP request | Description
  - [RunCreate](docs/RunCreate.md)
  - [RunSearchRequest](docs/RunSearchRequest.md)
  - [RunStatus](docs/RunStatus.md)
+ - [RunStream](docs/RunStream.md)
  - [RunWaitResponse](docs/RunWaitResponse.md)
  - [SearchAgentsRequest](docs/SearchAgentsRequest.md)
  - [SearchItemsResponse](docs/SearchItemsResponse.md)

client-python/ap_client/__init__.py

@@ -36,7 +36,7 @@
 # import models into sdk package
 from ap_client.models.agent import Agent
 from ap_client.models.agent_capabilities import AgentCapabilities
-from ap_client.models.agent_schemas import AgentSchemas
+from ap_client.models.agent_schema import AgentSchema
 from ap_client.models.config import Config
 from ap_client.models.content import Content
 from ap_client.models.content_one_of_inner import ContentOneOfInner
@@ -50,6 +50,7 @@
 from ap_client.models.run_create import RunCreate
 from ap_client.models.run_search_request import RunSearchRequest
 from ap_client.models.run_status import RunStatus
+from ap_client.models.run_stream import RunStream
 from ap_client.models.run_wait_response import RunWaitResponse
 from ap_client.models.search_agents_request import SearchAgentsRequest
 from ap_client.models.search_items_response import SearchItemsResponse

client-python/ap_client/api/agents_api.py

@@ -16,7 +16,7 @@
 from typing_extensions import Annotated
 
 from ap_client.models.agent import Agent
-from ap_client.models.agent_schemas import AgentSchemas
+from ap_client.models.agent_schema import AgentSchema
 from ap_client.models.search_agents_request import SearchAgentsRequest
 
 from ap_client.api_client import ApiClient, RequestSerialized
@@ -295,7 +295,7 @@ def get_agent_schemas(
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> AgentSchemas:
+    ) -> AgentSchema:
         """Get Agent Schemas
 
         Get an agent's schemas by ID.
@@ -333,7 +333,7 @@ def get_agent_schemas(
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "AgentSchemas",
+            "200": "AgentSchema",
             "404": "ErrorResponse",
             "422": "ErrorResponse",
         }
@@ -361,7 +361,7 @@ def get_agent_schemas_with_http_info(
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[AgentSchemas]:
+    ) -> ApiResponse[AgentSchema]:
         """Get Agent Schemas
 
         Get an agent's schemas by ID.
@@ -399,7 +399,7 @@ def get_agent_schemas_with_http_info(
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "AgentSchemas",
+            "200": "AgentSchema",
             "404": "ErrorResponse",
             "422": "ErrorResponse",
         }
@@ -465,7 +465,7 @@ def get_agent_schemas_without_preload_content(
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "AgentSchemas",
+            "200": "AgentSchema",
             "404": "ErrorResponse",
             "422": "ErrorResponse",
         }

client-python/ap_client/api/background_runs_api.py

@@ -17,8 +17,8 @@
 
 from pydantic import StrictBool
 from ap_client.models.run import Run
-from ap_client.models.run_create import RunCreate
 from ap_client.models.run_search_request import RunSearchRequest
+from ap_client.models.run_stream import RunStream
 from ap_client.models.run_wait_response import RunWaitResponse
 
 from ap_client.api_client import ApiClient, RequestSerialized
@@ -332,7 +332,7 @@ def _cancel_run_serialize(
     @validate_call
     def create_run(
         self,
-        run_create: RunCreate,
+        run_stream: RunStream,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -349,8 +349,8 @@ def create_run(
 
         Create a run in a new thread, return the run ID immediately. Don't wait for the final run output.
 
-        :param run_create: (required)
-        :type run_create: RunCreate
+        :param run_stream: (required)
+        :type run_stream: RunStream
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -374,7 +374,7 @@ def create_run(
         """  # noqa: E501
 
         _param = self._create_run_serialize(
-            run_create=run_create,
+            run_stream=run_stream,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -399,7 +399,7 @@ def create_run(
     @validate_call
     def create_run_with_http_info(
         self,
-        run_create: RunCreate,
+        run_stream: RunStream,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -416,8 +416,8 @@ def create_run_with_http_info(
 
         Create a run in a new thread, return the run ID immediately. Don't wait for the final run output.
 
-        :param run_create: (required)
-        :type run_create: RunCreate
+        :param run_stream: (required)
+        :type run_stream: RunStream
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -441,7 +441,7 @@ def create_run_with_http_info(
         """  # noqa: E501
 
         _param = self._create_run_serialize(
-            run_create=run_create,
+            run_stream=run_stream,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -466,7 +466,7 @@ def create_run_with_http_info(
     @validate_call
     def create_run_without_preload_content(
         self,
-        run_create: RunCreate,
+        run_stream: RunStream,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -483,8 +483,8 @@ def create_run_without_preload_content(
 
         Create a run in a new thread, return the run ID immediately. Don't wait for the final run output.
 
-        :param run_create: (required)
-        :type run_create: RunCreate
+        :param run_stream: (required)
+        :type run_stream: RunStream
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -508,7 +508,7 @@ def create_run_without_preload_content(
         """  # noqa: E501
 
         _param = self._create_run_serialize(
-            run_create=run_create,
+            run_stream=run_stream,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -528,7 +528,7 @@ def create_run_without_preload_content(
 
     def _create_run_serialize(
         self,
-        run_create,
+        run_stream,
         _request_auth,
         _content_type,
         _headers,
@@ -552,8 +552,8 @@ def _create_run_serialize(
         # process the header parameters
         # process the form parameters
         # process the body parameter
-        if run_create is not None:
-            _body_params = run_create
+        if run_stream is not None:
+            _body_params = run_stream
 
         # set the HTTP header `Accept`
         if "Accept" not in _header_params:

client-python/ap_client/api/runs_api.py

@@ -16,6 +16,7 @@
 from typing_extensions import Annotated
 
 from ap_client.models.run_create import RunCreate
+from ap_client.models.run_stream import RunStream
 from ap_client.models.run_wait_response import RunWaitResponse
 
 from ap_client.api_client import ApiClient, RequestSerialized
@@ -38,7 +39,7 @@ def __init__(self, api_client=None) -> None:
     @validate_call
     def create_and_stream_run(
         self,
-        run_create: RunCreate,
+        run_stream: RunStream,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -55,8 +56,8 @@ def create_and_stream_run(
 
         Create a run in a new thread, stream the output.
 
-        :param run_create: (required)
-        :type run_create: RunCreate
+        :param run_stream: (required)
+        :type run_stream: RunStream
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -80,7 +81,7 @@ def create_and_stream_run(
         """  # noqa: E501
 
         _param = self._create_and_stream_run_serialize(
-            run_create=run_create,
+            run_stream=run_stream,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -105,7 +106,7 @@ def create_and_stream_run(
     @validate_call
     def create_and_stream_run_with_http_info(
         self,
-        run_create: RunCreate,
+        run_stream: RunStream,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -122,8 +123,8 @@ def create_and_stream_run_with_http_info(
 
         Create a run in a new thread, stream the output.
 
-        :param run_create: (required)
-        :type run_create: RunCreate
+        :param run_stream: (required)
+        :type run_stream: RunStream
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -147,7 +148,7 @@ def create_and_stream_run_with_http_info(
         """  # noqa: E501
 
         _param = self._create_and_stream_run_serialize(
-            run_create=run_create,
+            run_stream=run_stream,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -172,7 +173,7 @@ def create_and_stream_run_with_http_info(
     @validate_call
     def create_and_stream_run_without_preload_content(
         self,
-        run_create: RunCreate,
+        run_stream: RunStream,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -189,8 +190,8 @@ def create_and_stream_run_without_preload_content(
 
         Create a run in a new thread, stream the output.
 
-        :param run_create: (required)
-        :type run_create: RunCreate
+        :param run_stream: (required)
+        :type run_stream: RunStream
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -214,7 +215,7 @@ def create_and_stream_run_without_preload_content(
         """  # noqa: E501
 
         _param = self._create_and_stream_run_serialize(
-            run_create=run_create,
+            run_stream=run_stream,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -234,7 +235,7 @@ def create_and_stream_run_without_preload_content(
 
     def _create_and_stream_run_serialize(
         self,
-        run_create,
+        run_stream,
         _request_auth,
         _content_type,
         _headers,
@@ -258,8 +259,8 @@ def _create_and_stream_run_serialize(
         # process the header parameters
         # process the form parameters
         # process the body parameter
-        if run_create is not None:
-            _body_params = run_create
+        if run_stream is not None:
+            _body_params = run_stream
 
         # set the HTTP header `Accept`
         if "Accept" not in _header_params:

client-python/ap_client/models/__init__.py

@@ -15,7 +15,7 @@
 # import models into model package
 from ap_client.models.agent import Agent
 from ap_client.models.agent_capabilities import AgentCapabilities
-from ap_client.models.agent_schemas import AgentSchemas
+from ap_client.models.agent_schema import AgentSchema
 from ap_client.models.config import Config
 from ap_client.models.content import Content
 from ap_client.models.content_one_of_inner import ContentOneOfInner
@@ -29,6 +29,7 @@
 from ap_client.models.run_create import RunCreate
 from ap_client.models.run_search_request import RunSearchRequest
 from ap_client.models.run_status import RunStatus
+from ap_client.models.run_stream import RunStream
 from ap_client.models.run_wait_response import RunWaitResponse
 from ap_client.models.search_agents_request import SearchAgentsRequest
 from ap_client.models.search_items_response import SearchItemsResponse