Release v0.1.69

superagent-ai · Jan 30, 2024 · 182e1d0 · 182e1d0
1 parent 53be2ce
commit 182e1d0
Show file tree

Hide file tree

Showing 13 changed files with 288 additions and 34 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "superagent-py"
-version = "v0.1.68"
+version = "v0.1.69"
 description = ""
 readme = "README.md"
 authors = []

diff --git a/src/superagent/__init__.py b/src/superagent/__init__.py
@@ -5,7 +5,6 @@
     AgentList,
     AgentRunList,
     AgentToolList,
-    AppModelsRequestAgent,
     AppModelsRequestDatasource,
     AppModelsRequestLlm,
     AppModelsRequestTool,
@@ -46,19 +45,19 @@
     ValidationErrorLocItem,
     VectorDbList,
     VectorDbProvider,
+    WorkflowConfig,
     WorkflowList,
     WorkflowStepList,
 )
 from .errors import UnprocessableEntityError
-from .resources import agent, api_user, datasource, llm, telemetry, tool, vector_database, workflow
+from .resources import agent, api_user, datasource, llm, telemetry, tool, vector_database, workflow, workflow_config
 from .environment import SuperagentEnvironment
 
 __all__ = [
     "AgentDatasosurceList",
     "AgentList",
     "AgentRunList",
     "AgentToolList",
-    "AppModelsRequestAgent",
     "AppModelsRequestDatasource",
     "AppModelsRequestLlm",
     "AppModelsRequestTool",
@@ -101,6 +100,7 @@
     "ValidationErrorLocItem",
     "VectorDbList",
     "VectorDbProvider",
+    "WorkflowConfig",
     "WorkflowList",
     "WorkflowStepList",
     "agent",
@@ -111,4 +111,5 @@
     "tool",
     "vector_database",
     "workflow",
+    "workflow_config",
 ]
diff --git a/src/superagent/client.py b/src/superagent/client.py
@@ -14,6 +14,7 @@
 from .resources.tool.client import AsyncToolClient, ToolClient
 from .resources.vector_database.client import AsyncVectorDatabaseClient, VectorDatabaseClient
 from .resources.workflow.client import AsyncWorkflowClient, WorkflowClient
+from .resources.workflow_config.client import AsyncWorkflowConfigClient, WorkflowConfigClient
 
 
 class Superagent:
@@ -37,6 +38,7 @@ def __init__(
         self.datasource = DatasourceClient(client_wrapper=self._client_wrapper)
         self.tool = ToolClient(client_wrapper=self._client_wrapper)
         self.workflow = WorkflowClient(client_wrapper=self._client_wrapper)
+        self.workflow_config = WorkflowConfigClient(client_wrapper=self._client_wrapper)
         self.vector_database = VectorDatabaseClient(client_wrapper=self._client_wrapper)
         self.telemetry = TelemetryClient(client_wrapper=self._client_wrapper)
 
@@ -62,6 +64,7 @@ def __init__(
         self.datasource = AsyncDatasourceClient(client_wrapper=self._client_wrapper)
         self.tool = AsyncToolClient(client_wrapper=self._client_wrapper)
         self.workflow = AsyncWorkflowClient(client_wrapper=self._client_wrapper)
+        self.workflow_config = AsyncWorkflowConfigClient(client_wrapper=self._client_wrapper)
         self.vector_database = AsyncVectorDatabaseClient(client_wrapper=self._client_wrapper)
         self.telemetry = AsyncTelemetryClient(client_wrapper=self._client_wrapper)
 

diff --git a/src/superagent/core/client_wrapper.py b/src/superagent/core/client_wrapper.py
@@ -14,7 +14,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "superagent-py",
-            "X-Fern-SDK-Version": "v0.1.68",
+            "X-Fern-SDK-Version": "v0.1.69",
         }
         token = self._get_token()
         if token is not None:

diff --git a/src/superagent/resources/__init__.py b/src/superagent/resources/__init__.py
@@ -1,5 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from . import agent, api_user, datasource, llm, telemetry, tool, vector_database, workflow
+from . import agent, api_user, datasource, llm, telemetry, tool, vector_database, workflow, workflow_config
 
-__all__ = ["agent", "api_user", "datasource", "llm", "telemetry", "tool", "vector_database", "workflow"]
+__all__ = [
+    "agent",
+    "api_user",
+    "datasource",
+    "llm",
+    "telemetry",
+    "tool",
+    "vector_database",
+    "workflow",
+    "workflow_config",
+]
diff --git a/src/superagent/resources/agent/client.py b/src/superagent/resources/agent/client.py
@@ -12,11 +12,11 @@
 from ...types.agent_datasosurce_list import AgentDatasosurceList
 from ...types.agent_list import AgentList
 from ...types.agent_tool_list import AgentToolList
-from ...types.app_models_request_agent import AppModelsRequestAgent
 from ...types.app_models_response_agent import AppModelsResponseAgent
 from ...types.app_models_response_agent_invoke import AppModelsResponseAgentInvoke
 from ...types.http_validation_error import HttpValidationError
 from ...types.llm_params import LlmParams
+from ...types.llm_provider import LlmProvider
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -57,17 +57,57 @@ def list(self, *, skip: typing.Optional[int] = None, take: typing.Optional[int]
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create(self, *, request: AppModelsRequestAgent) -> AppModelsResponseAgent:
+    def create(
+        self,
+        *,
+        is_active: typing.Optional[bool] = OMIT,
+        name: str,
+        initial_message: typing.Optional[str] = OMIT,
+        prompt: typing.Optional[str] = OMIT,
+        llm_model: typing.Optional[str] = OMIT,
+        llm_provider: typing.Optional[LlmProvider] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        avatar: typing.Optional[str] = OMIT,
+    ) -> AppModelsResponseAgent:
         """
         Create a new agent
 
         Parameters:
-            - request: AppModelsRequestAgent.
+            - is_active: typing.Optional[bool].
+
+            - name: str.
+
+            - initial_message: typing.Optional[str].
+
+            - prompt: typing.Optional[str].
+
+            - llm_model: typing.Optional[str].
+
+            - llm_provider: typing.Optional[LlmProvider].
+
+            - description: typing.Optional[str].
+
+            - avatar: typing.Optional[str].
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name}
+        if is_active is not OMIT:
+            _request["isActive"] = is_active
+        if initial_message is not OMIT:
+            _request["initialMessage"] = initial_message
+        if prompt is not OMIT:
+            _request["prompt"] = prompt
+        if llm_model is not OMIT:
+            _request["llmModel"] = llm_model
+        if llm_provider is not OMIT:
+            _request["llmProvider"] = llm_provider.value
+        if description is not OMIT:
+            _request["description"] = description
+        if avatar is not OMIT:
+            _request["avatar"] = avatar
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/agents"),
-            json=jsonable_encoder(request),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -127,19 +167,57 @@ def delete(self, agent_id: str) -> typing.Any:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def update(self, agent_id: str, *, request: AppModelsRequestAgent) -> AppModelsResponseAgent:
+    def update(
+        self,
+        agent_id: str,
+        *,
+        is_active: typing.Optional[bool] = OMIT,
+        name: typing.Optional[str] = OMIT,
+        initial_message: typing.Optional[str] = OMIT,
+        prompt: typing.Optional[str] = OMIT,
+        llm_model: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        avatar: typing.Optional[str] = OMIT,
+    ) -> AppModelsResponseAgent:
         """
         Patch an agent
 
         Parameters:
             - agent_id: str.
 
-            - request: AppModelsRequestAgent.
+            - is_active: typing.Optional[bool].
+
+            - name: typing.Optional[str].
+
+            - initial_message: typing.Optional[str].
+
+            - prompt: typing.Optional[str].
+
+            - llm_model: typing.Optional[str].
+
+            - description: typing.Optional[str].
+
+            - avatar: typing.Optional[str].
         """
+        _request: typing.Dict[str, typing.Any] = {}
+        if is_active is not OMIT:
+            _request["isActive"] = is_active
+        if name is not OMIT:
+            _request["name"] = name
+        if initial_message is not OMIT:
+            _request["initialMessage"] = initial_message
+        if prompt is not OMIT:
+            _request["prompt"] = prompt
+        if llm_model is not OMIT:
+            _request["llmModel"] = llm_model
+        if description is not OMIT:
+            _request["description"] = description
+        if avatar is not OMIT:
+            _request["avatar"] = avatar
         _response = self._client_wrapper.httpx_client.request(
             "PATCH",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/agents/{agent_id}"),
-            json=jsonable_encoder(request),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -437,17 +515,57 @@ async def list(self, *, skip: typing.Optional[int] = None, take: typing.Optional
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def create(self, *, request: AppModelsRequestAgent) -> AppModelsResponseAgent:
+    async def create(
+        self,
+        *,
+        is_active: typing.Optional[bool] = OMIT,
+        name: str,
+        initial_message: typing.Optional[str] = OMIT,
+        prompt: typing.Optional[str] = OMIT,
+        llm_model: typing.Optional[str] = OMIT,
+        llm_provider: typing.Optional[LlmProvider] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        avatar: typing.Optional[str] = OMIT,
+    ) -> AppModelsResponseAgent:
         """
         Create a new agent
 
         Parameters:
-            - request: AppModelsRequestAgent.
+            - is_active: typing.Optional[bool].
+
+            - name: str.
+
+            - initial_message: typing.Optional[str].
+
+            - prompt: typing.Optional[str].
+
+            - llm_model: typing.Optional[str].
+
+            - llm_provider: typing.Optional[LlmProvider].
+
+            - description: typing.Optional[str].
+
+            - avatar: typing.Optional[str].
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name}
+        if is_active is not OMIT:
+            _request["isActive"] = is_active
+        if initial_message is not OMIT:
+            _request["initialMessage"] = initial_message
+        if prompt is not OMIT:
+            _request["prompt"] = prompt
+        if llm_model is not OMIT:
+            _request["llmModel"] = llm_model
+        if llm_provider is not OMIT:
+            _request["llmProvider"] = llm_provider.value
+        if description is not OMIT:
+            _request["description"] = description
+        if avatar is not OMIT:
+            _request["avatar"] = avatar
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/agents"),
-            json=jsonable_encoder(request),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -507,19 +625,57 @@ async def delete(self, agent_id: str) -> typing.Any:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def update(self, agent_id: str, *, request: AppModelsRequestAgent) -> AppModelsResponseAgent:
+    async def update(
+        self,
+        agent_id: str,
+        *,
+        is_active: typing.Optional[bool] = OMIT,
+        name: typing.Optional[str] = OMIT,
+        initial_message: typing.Optional[str] = OMIT,
+        prompt: typing.Optional[str] = OMIT,
+        llm_model: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        avatar: typing.Optional[str] = OMIT,
+    ) -> AppModelsResponseAgent:
         """
         Patch an agent
 
         Parameters:
             - agent_id: str.
 
-            - request: AppModelsRequestAgent.
+            - is_active: typing.Optional[bool].
+
+            - name: typing.Optional[str].
+
+            - initial_message: typing.Optional[str].
+
+            - prompt: typing.Optional[str].
+
+            - llm_model: typing.Optional[str].
+
+            - description: typing.Optional[str].
+
+            - avatar: typing.Optional[str].
         """
+        _request: typing.Dict[str, typing.Any] = {}
+        if is_active is not OMIT:
+            _request["isActive"] = is_active
+        if name is not OMIT:
+            _request["name"] = name
+        if initial_message is not OMIT:
+            _request["initialMessage"] = initial_message
+        if prompt is not OMIT:
+            _request["prompt"] = prompt
+        if llm_model is not OMIT:
+            _request["llmModel"] = llm_model
+        if description is not OMIT:
+            _request["description"] = description
+        if avatar is not OMIT:
+            _request["avatar"] = avatar
         _response = await self._client_wrapper.httpx_client.request(
             "PATCH",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/agents/{agent_id}"),
-            json=jsonable_encoder(request),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )

diff --git a/src/superagent/resources/workflow_config/__init__.py b/src/superagent/resources/workflow_config/__init__.py
@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+