codemie-sdk-python 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

codemie_sdk/models/assistant.py

@@ -177,6 +177,30 @@ class Context(BaseModel):
     name: str
 
 
+class MCPServerConfig(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+
+    command: str
+    args: Optional[list[str]]
+    env: Optional[dict[str, Any]] = None
+    auth_token: Optional[str] = None
+
+
+class MCPServerDetails(BaseModel):
+    model_config = ConfigDict(extra="ignore")
+
+    name: str
+    description: Optional[str]
+    enabled: bool
+    config: Optional[MCPServerConfig] = None
+    mcp_connect_url: Optional[str] = None
+    tools_tokens_size_limit: Optional[int] = None
+    command: Optional[str] = None
+    arguments: Optional[str] = None
+    settings: Optional[Integration] = None
+    mcp_connect_auth_token: Optional[Integration] = None
+
+
 class SystemPromptHistory(BaseModel):
     """Model for system prompt history."""
 
@@ -221,6 +245,8 @@ class Assistant(AssistantBase):
     top_p: Optional[float] = None
     context: List[Context] = Field(default_factory=list)
     user_abilities: Optional[List[Any]] = None
+    mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
+    assistant_ids: List[str] = Field(default_factory=list)
 
 
 class AssistantRequestBase(AssistantBase):
@@ -232,7 +258,7 @@ class AssistantRequestBase(AssistantBase):
     project: str
     context: List[Context] = Field(default_factory=list)
     llm_model_type: str
-    toolkits: List[ToolKitDetails]
+    toolkits: List[ToolKitDetails] = Field(default_factory=list)
     user_prompts: List[str] = Field(default_factory=list)
     shared: bool = False
     is_react: bool = False
@@ -240,6 +266,8 @@ class AssistantRequestBase(AssistantBase):
     slug: Optional[str] = None
     temperature: Optional[float] = None
     top_p: Optional[float] = None
+    mcp_servers: List[MCPServerDetails] = Field(default_factory=list)
+    assistant_ids: List[str] = Field(default_factory=list)
 
 
 class AssistantCreateRequest(AssistantRequestBase):

{codemie_sdk_python-0.1.10.dist-info → codemie_sdk_python-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.10
+Version: 0.1.12
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com
@@ -57,6 +57,12 @@ Python SDK for CodeMie services. This SDK provides a comprehensive interface to
 ```sh
 pip install codemie-sdk
 ```
+OR
+```sh
+poetry install
+```
+
+### If you want to run only tests, go to ## Running tests section
 
 ## Usage
 
@@ -740,7 +746,8 @@ For providing credentials please contact AI/Run CodeMie Team: Vadym_Vlasenko@epa
 
 ## Running tests
 
-For running tests on custom environment you should create .env file in the root directory
+For running tests on custom environment you should create .env file in the root directory,
+ask QA team: anton_yeromin@epam.com to provide all needed testing credentials
 
 ``` properties
 

{codemie_sdk_python-0.1.10.dist-info → codemie_sdk_python-0.1.12.dist-info}/RECORD

@@ -4,7 +4,7 @@ codemie_sdk/auth/credentials.py,sha256=u2eLNsD8fELTgreQghVb0g0kk82zchUkux0M5lzq-
 codemie_sdk/client/__init__.py,sha256=yf6C39MmrJ6gK9ZHMhBeynKwUUYVSUTQbKxU8-4qpKg,101
 codemie_sdk/client/client.py,sha256=dVcWqd-ruWd8GOZ3_33C3LnHgUyyxIoCKxKYPYbBPq8,4373
 codemie_sdk/exceptions.py,sha256=XoVPyognx-JmyVxLHkZPAcX1CMi1OoT1diBFJLU54so,1183
-codemie_sdk/models/assistant.py,sha256=yYJMEj7InpM3Wt321qhQ-m8PLBwayIhp6rZqS3HHmyA,9028
+codemie_sdk/models/assistant.py,sha256=IgTu1OnoWnoDVA0pA8IQkNtpgypgSUzkLE9tfzoBJKs,9978
 codemie_sdk/models/common.py,sha256=V4sJCzwFTF8BPlrd2OKI44Em5HmPmn2Nm8fQNsnBZ_Q,1128
 codemie_sdk/models/datasource.py,sha256=Iun6O76o4m-_jbyyumMndpyr4OiN0lwqKb8u5AOTHdM,10037
 codemie_sdk/models/integration.py,sha256=F1YBdCK1w5W_jKK8prEsg7A00NghkEBvJgkkr8KpC6w,1764
@@ -24,6 +24,6 @@ codemie_sdk/services/workflow_execution.py,sha256=aGoT3rdTmh5-doAsrmBBjLEuOfvL5a
 codemie_sdk/services/workflow_execution_state.py,sha256=tXoaa8yT09xgYEUNiHhVULe76TwGwVgZupMIUyyLxdo,2070
 codemie_sdk/utils/__init__.py,sha256=BXAJJfAzO89-kMYvWWo9wSNhSbGgF3vB1In9sePFhMM,109
 codemie_sdk/utils/http.py,sha256=JfyX4gP-grQUI795QL0DJ72aBkiqNd1fs31fUqtSnYE,7757
-codemie_sdk_python-0.1.10.dist-info/METADATA,sha256=NtLyms1ufzGRn6EpQHMYKFpiguunPVhE_C9n2T_X7Tk,21440
-codemie_sdk_python-0.1.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-codemie_sdk_python-0.1.10.dist-info/RECORD,,
+codemie_sdk_python-0.1.12.dist-info/METADATA,sha256=uRl1oD_4pvVqaGchWkwjNg_tWw0_5rKIscA2b4RY7ts,21614
+codemie_sdk_python-0.1.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+codemie_sdk_python-0.1.12.dist-info/RECORD,,