sparkContext 0.1.0__tar.gz

sparkcontext-0.1.0/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: sparkContext
+Version: 0.1.0
+Author: DeepModel
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Requires-Dist: httpx>=0.24.0
+Dynamic: author
+Dynamic: classifier
+Dynamic: requires-dist
+Dynamic: requires-python

sparkcontext-0.1.0/README.md

@@ -0,0 +1,251 @@
+# Spark SDK
+
+A Python SDK for interacting with the DeepModel Spark API.
+
+---
+
+## Installation
+
+### Install from PyPI
+```bash
+pip install spark-sdk
+```
+
+### Install from Source
+```bash
+# 1. Clone the repository
+git clone https://github.com/your-org/dm-spark-sdk.git
+cd dm-spark-sdk
+
+# 2. Install dependencies
+pip install setuptools wheel twine httpx
+
+# 3. Build the package
+python setup.py sdist bdist_wheel
+
+# 4. Install the built package
+pip install dist\spark_sdk-0.1.0-py3-none-any.whl
+
+# Upload to PyPI
+twine upload dist/*
+```
+
+---
+
+## Quick Start
+
+```python
+import asyncio
+import spark_sdk
+
+async def main():
+    # Initialize the client
+    client = spark_sdk.Client(
+        api_key="your_api_key",
+        api_url="https://your-api-url.com"
+    )
+
+    # Get all unified tools and agents
+    result = await client.get_unified_tools()
+    print(result)
+
+    # Invoke an MCP tool
+    invoke_result = await client.invoke_tool(
+        connection_id="uuid-here",
+        target_type=spark_sdk.MCP,
+        tool_or_agent_name="create_issue",
+        arguments={"title": "Bug report", "body": "Something went wrong"},
+        query="create a github issue"
+    )
+    print(invoke_result)
+
+asyncio.run(main())
+```
+
+---
+
+## Usage
+
+### Initialize the Client
+
+```python
+import spark_sdk
+
+client = spark_sdk.Client(
+    api_key="your_api_key",
+    api_url="https://your-api-url.com"
+)
+```
+
+| Parameter | Type   | Required | Description                        |
+|-----------|--------|----------|------------------------------------|
+| `api_key` | `str`  | ✅ Yes   | Your API key for authentication    |
+| `api_url` | `str`  | ✅ Yes   | The base URL of the Spark API      |
+
+---
+
+### `get_unified_tools()`
+
+Retrieves all MCP tools and worker agents accessible to the user, including their connection status and payload schemas.
+
+```python
+result = await client.get_unified_tools()
+```
+
+#### Response Structure
+
+```json
+{
+    "success": true,
+    "tools": {
+        "mcp_tools": {
+            "github": [
+                {
+                    "tool_name": "create_issue",
+                    "is_configured": true,
+                    "connection_id": "uuid-here",
+                    "description": "Creates a GitHub issue",
+                    "payload_schema": {},
+                    "icon_url": "https://..."
+                }
+            ]
+        },
+        "worker_agents": [
+            {
+                "tool_name": "My Agent",
+                "is_configured": true,
+                "connection_id": "uuid-here",
+                "description": "A custom worker agent",
+                "payload_schema": {},
+                "agent_avatar_url": "https://..."
+            }
+        ]
+    },
+    "status_code": 200,
+    "detail": null
+}
+```
+
+| Field         | Type           | Description                                  |
+|---------------|----------------|----------------------------------------------|
+| `success`     | `bool`         | Whether the request was successful           |
+| `tools`       | `dict`         | All MCP tools and worker agents              |
+| `status_code` | `int`          | HTTP status code of the response             |
+| `detail`      | `str` / `null` | Error detail message if request failed       |
+
+---
+
+### `invoke_tool()`
+
+Directly invoke an MCP tool or worker agent with provided arguments. No LLM inference is performed — arguments are used as-is.
+
+```python
+result = await client.invoke_tool(
+    connection_id="uuid-here",
+    target_type=spark_sdk.MCP,         # or spark_sdk.AGENT
+    tool_or_agent_name="create_issue",
+    arguments={"title": "Bug report", "body": "Something went wrong"},
+    query="create a github issue"      # optional
+)
+```
+
+#### Parameters
+
+| Parameter            | Type            | Required | Description                                          |
+|----------------------|-----------------|----------|------------------------------------------------------|
+| `connection_id`      | `str` / `UUID`  | ✅ Yes   | The connection ID of the tool or agent               |
+| `target_type`        | `str`           | ✅ Yes   | `spark_sdk.MCP` (`"mcp"`) or `spark_sdk.AGENT` (`"agent"`) |
+| `tool_or_agent_name` | `str`           | ✅ Yes   | The name of the tool or agent to invoke              |
+| `arguments`          | `dict`          | ✅ Yes   | Arguments to pass to the tool or agent               |
+| `query`              | `str`           | ❌ No    | Optional user query context for agent invocation     |
+
+#### Response Structure
+
+```json
+{
+    "success": true,
+    "tool_or_agent_name": "create_issue",
+    "connection_id": "uuid-here",
+    "result": {},
+    "status_code": 200,
+    "detail": null
+}
+```
+
+| Field                | Type           | Description                                       |
+|----------------------|----------------|---------------------------------------------------|
+| `success`            | `bool`         | Whether the invocation was successful             |
+| `tool_or_agent_name` | `str`          | The name of the tool or agent invoked             |
+| `connection_id`      | `str`          | The connection ID used                            |
+| `result`             | `any`          | The result returned by the tool or agent          |
+| `status_code`        | `int`          | HTTP status code of the response                  |
+| `detail`             | `str` / `null` | Error detail message if invocation failed         |
+
+#### Target Type Constants
+
+```python
+spark_sdk.MCP    # "mcp"   — for MCP tools
+spark_sdk.AGENT  # "agent" — for worker agents
+```
+
+---
+
+### Error Handling
+
+All errors are returned as part of the response dictionary rather than raising exceptions:
+
+```python
+result = await client.get_unified_tools()
+
+if not result.get("success"):
+    print(f"Error {result.get('status_code')}: {result.get('detail')}")
+else:
+    print(result.get("tools"))
+```
+
+| Error Type       | `status_code` | `detail` Example                                     |
+|------------------|---------------|------------------------------------------------------|
+| HTTP Error       | e.g. `401`    | `"No detail provided"`                               |
+| Request Error    | `400`         | `"Request error: Connection refused"`                |
+| Unexpected Error | `500`         | `"An unexpected error occurred: ..."`                |
+
+---
+
+## Project Structure
+
+```
+dm-spark-sdk/
+├── spark_sdk/
+│   ├── __init__.py       # Exposes Client
+│   └── client.py         # SparkClient class with all methods
+├── setup.py              # Package configuration
+└── README.md
+```
+
+---
+
+## Requirements
+
+- Python `>= 3.8`
+- `httpx >= 0.24.0`
+
+---
+
+## Publishing to PyPI
+
+```bash
+# Install twine
+pip install twine
+
+# Build the package
+python setup.py sdist bdist_wheel
+
+# Upload to PyPI
+twine upload dist/*
+```
+
+---
+
+## License
+
+This project is licensed under the MIT License.

sparkcontext-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

sparkcontext-0.1.0/setup.py

@@ -0,0 +1,17 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='sparkContext',
+    version='0.1.0',
+    author='DeepModel',
+    packages=find_packages(),
+    install_requires=[
+        'httpx>=0.24.0',
+    ],
+    python_requires='>=3.8',
+    classifiers=[
+        'Programming Language :: Python :: 3',
+        'License :: OSI Approved :: MIT License',
+        'Operating System :: OS Independent',
+    ],
+)

sparkcontext-0.1.0/sparkContext/__init__.py

@@ -0,0 +1,9 @@
+from .client import SparkClient
+
+Client = SparkClient
+
+# Target type constants for invoke_tool
+MCP = "mcp"
+AGENT = "agent"
+
+__all__ = ["Client", "SparkClient", "MCP", "AGENT"]

sparkcontext-0.1.0/sparkContext/client.py

@@ -0,0 +1,166 @@
+import httpx
+from typing import Any, Dict, Optional, Union
+from uuid import UUID
+
+
+class SparkClient:
+    """
+    Spark SDK Client.
+    
+    Usage:
+        import sparkContext
+        client = sparkContext.Client(api_key="your_api_key", api_url="https://your-api-url.com")
+        result = await client.get_tools()
+    """
+    def __init__(self, api_key: str, api_url: str):
+        """
+        Initialize the Spark SDK Client.
+
+        Args:
+            api_key (str): Your API key for authentication.
+            api_url (str): The base URL of the Spark API.
+        """
+        if not api_key:
+            raise ValueError("API key is required.")
+        if not api_url:
+            raise ValueError("API URL is required.")
+        
+        self.api_key = api_key
+        self.api_url = api_url.rstrip('/')
+        self.headers = {
+            "x-api-key": self.api_key,
+            "Content-Type": "application/json"
+        }
+
+
+    async def get_unified_tools(self):
+        """
+        Get all tools available in the Spark.
+        Returns all available tools along with connected and not connected tools.
+
+        Returns:
+            dict: A dictionary containing all available tools along with connected and not connected tools.
+        """
+        endpoint = f"{self.api_url}/utils/unified-tools-and-agents"
+
+        async with httpx.AsyncClient(timeout=None) as client:
+            try:
+                response = await client.get(endpoint, headers=self.headers)
+                response.raise_for_status()
+                return {
+                    "success": True,
+                    "tools": response.json(),
+                    "status_code": response.status_code,
+                    "detail": None
+                }
+            except httpx.HTTPStatusError as e:
+                return {
+                    "success": False,
+                    "tools": None,
+                    "status_code": e.response.status_code,
+                    "detail": e.response.json().get("detail", "No detail provided")
+                }
+            except httpx.RequestError as e:
+                return {
+                    "success": False,
+                    "tools": None,
+                    "status_code": 400,
+                    "detail": f"An error occurred while making the request: {str(e)}"
+                }
+            except Exception as e:
+                return {
+                    "success": False,
+                    "tools": None,
+                    "status_code": 500,
+                    "detail": f"An unexpected error occurred: {str(e)}"
+                }
+            
+    
+    async def invoke_tool(self, connection_id: Union[str, UUID], target_type: str, tool_or_agent_name: str, arguments: Dict[str, Any], query: Optional[str] = "") -> dict:
+        """
+        Directly invoke an MCP tool or worker agent with provided arguments.
+        No LLM inference is performed — arguments are used as-is.
+        Required parameters are validated before execution.
+
+        Args:
+            connection_id (str): The connection ID of the tool or agent.
+            target_type (str): The type of target to invoke. Either "mcp" or "agent".
+            tool_or_agent_name (str): The name of the tool or agent to invoke.
+            arguments (Dict[str, Any]): The arguments to pass to the tool or agent.
+            query (Optional[str]): Optional user query context for agent invocation.
+
+        Returns:
+            dict: A dictionary containing:
+                - success (bool): Whether the invocation was successful
+                - tool_or_agent_name (str): The name of the tool or agent invoked
+                - connection_id (str): The connection ID of the tool or agent invoked
+                - result (Any): The result of the invocation
+                - status_code (int): The HTTP status code of the response
+                - detail (Optional[str]): Additional error details if any
+
+        Example:
+            result = await client.invoke_tool(
+                connection_id="uuid-here",
+                target_type="mcp",
+                tool_or_agent_name="create_issue",
+                arguments={"title": "Bug report", "body": "Something went wrong"},
+                query="create a github issue"
+            )
+        """
+        if target_type not in ("mcp", "agent"):
+            return {
+                "success": False,
+                "tool_or_agent_name": tool_or_agent_name,
+                "connection_id": connection_id,
+                "result": None,
+                "status_code": 400,
+                "detail": "Invalid target_type. Must be 'mcp' or 'agent'."
+            }
+
+        endpoint = f"{self.api_url}/utils/tools/invoke"
+        params = {"query": query} if query else {}
+        payload = {
+            "connection_id": connection_id,
+            "target_type": target_type.lower(),
+            "tool_or_agent_name": tool_or_agent_name,
+            "arguments": arguments
+        }
+
+        async with httpx.AsyncClient(timeout=None) as client:
+            try:
+                response = await client.post(
+                    endpoint,
+                    headers=self.headers,
+                    json=payload,
+                    params=params
+                )
+                response.raise_for_status()
+                return response.json()
+
+            except httpx.HTTPStatusError as e:
+                return {
+                    "success": False,
+                    "tool_or_agent_name": tool_or_agent_name,
+                    "connection_id": connection_id,
+                    "result": None,
+                    "status_code": e.response.status_code,
+                    "detail": e.response.json().get("detail", "No detail provided")
+                }
+            except httpx.RequestError as e:
+                return {
+                    "success": False,
+                    "tool_or_agent_name": tool_or_agent_name,
+                    "connection_id": connection_id,
+                    "result": None,
+                    "status_code": 400,
+                    "detail": f"Request error: {str(e)}"
+                }
+            except Exception as e:
+                return {
+                    "success": False,
+                    "tool_or_agent_name": tool_or_agent_name,
+                    "connection_id": connection_id,
+                    "result": None,
+                    "status_code": 500,
+                    "detail": f"Unexpected error: {str(e)}"
+                }

sparkcontext-0.1.0/sparkContext.egg-info/PKG-INFO

@@ -0,0 +1,13 @@
+Metadata-Version: 2.4
+Name: sparkContext
+Version: 0.1.0
+Author: DeepModel
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Requires-Dist: httpx>=0.24.0
+Dynamic: author
+Dynamic: classifier
+Dynamic: requires-dist
+Dynamic: requires-python

sparkcontext-0.1.0/sparkContext.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+README.md
+setup.py
+sparkContext/__init__.py
+sparkContext/client.py
+sparkContext.egg-info/PKG-INFO
+sparkContext.egg-info/SOURCES.txt
+sparkContext.egg-info/dependency_links.txt
+sparkContext.egg-info/requires.txt
+sparkContext.egg-info/top_level.txt

sparkcontext-0.1.0/sparkContext.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

sparkcontext-0.1.0/sparkContext.egg-info/requires.txt

@@ -0,0 +1 @@
+httpx>=0.24.0

sparkcontext-0.1.0/sparkContext.egg-info/top_level.txt

@@ -0,0 +1 @@
+sparkContext