sunholo 0.115.2__tar.gz → 0.116.0__tar.gz

{sunholo-0.115.2 → sunholo-0.116.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.115.2
+Version: 0.116.0
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.115.2.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.116.0.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0
@@ -138,7 +138,8 @@ Provides-Extra: openai
 Requires-Dist: langchain-openai==0.1.25; extra == "openai"
 Requires-Dist: tiktoken; extra == "openai"
 Provides-Extra: anthropic
-Requires-Dist: langchain-anthropic==0.1.23; extra == "anthropic"
+Requires-Dist: langchain-anthropic>=0.1.23; extra == "anthropic"
+Requires-Dist: mcp; extra == "anthropic"
 Provides-Extra: tools
 Requires-Dist: openapi-spec-validator; extra == "tools"
 Requires-Dist: playwright; extra == "tools"

{sunholo-0.115.2 → sunholo-0.116.0}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-version = '0.115.2'
+version = '0.116.0'
 
 setup(
     name='sunholo',
@@ -159,7 +159,8 @@ setup(
             "tiktoken"
         ],
         'anthropic': [
-            "langchain-anthropic==0.1.23"   
+            "langchain-anthropic>=0.1.23",
+            "mcp"
         ],
         'tools' : [
             'openapi-spec-validator',
@@ -188,7 +189,8 @@ setup(
             'google-cloud-texttospeech',
             'numpy',
             'sounddevice',
-        ]
+        ],
+        
 
     },
     classifiers=[

{sunholo-0.115.2 → sunholo-0.116.0}/sunholo/cli/cli.py

@@ -14,6 +14,8 @@ from ..llamaindex import setup_llamaindex_subparser
 from ..excel import setup_excel_subparser
 from ..terraform import setup_tfvarseditor_subparser
 from ..senses.stream_voice import setup_tts_subparser
+from ..mcp.cli import setup_mcp_subparser
+
 
 from ..utils import ConfigManager
 from ..utils.version import sunholo_version
@@ -101,6 +103,8 @@ def main(args=None):
     setup_tfvarseditor_subparser(subparsers)
     # tts
     setup_tts_subparser(subparsers)
+    # anthropic MCP
+    setup_mcp_subparser(subparsers)
 
     #TODO: add database setup commands: alloydb and supabase
 

sunholo-0.116.0/sunholo/mcp/cli.py

@@ -0,0 +1,256 @@
+import os
+import asyncio
+from typing import Any, Sequence
+from functools import lru_cache
+import subprocess
+from mcp.server import Server
+from mcp.server.stdio import stdio_server
+from mcp.types import (
+    Resource,
+    Tool,
+    TextContent,
+    ImageContent,
+    EmbeddedResource,
+)
+"""
+try:
+    from mcp.server import Server
+    from mcp.server.stdio import stdio_server
+    from mcp.types import (
+        Resource,
+        Tool,
+        TextContent,
+        ImageContent,
+        EmbeddedResource,
+    )
+except ImportError:
+    Server = None
+"""
+
+from pydantic import AnyUrl
+
+# Configure logging
+from ..custom_logging import setup_logging
+logger = setup_logging("sunholo-mcp-server")
+
+class SunholoMCPServer:
+    def __init__(self):
+        if Server is None:
+            raise ImportError("SunholoMCPServer requires `sunholo[anthropic]` to be installed")
+        
+        self.server = Server("sunholo-mcp-server")
+        self.setup_handlers()
+
+    def setup_handlers(self):
+        """Set up all the MCP protocol handlers"""
+        self.setup_resource_handlers()
+        self.setup_tool_handlers()
+
+    def setup_resource_handlers(self):
+        """Configure resource-related handlers"""
+        
+        @self.server.list_resources()
+        async def list_resources() -> list[Resource]:
+            """List available Sunholo resources"""
+            return [
+                Resource(
+                    uri="sunholo://vacs/list",
+                    name="Available Sunholo VACs",
+                    mimeType="application/json",
+                    description="List of available Virtual Agent Computers"
+                )
+            ]
+
+        @self.server.read_resource()
+        async def read_resource(uri: AnyUrl) -> str:
+            """Read Sunholo resources based on URI"""
+            if str(uri) == "sunholo://vacs/list":
+                try:
+                    # Execute sunholo vac list command
+                    result = subprocess.run(
+                        ["sunholo", "vac", "list", "--debug"],
+                        capture_output=True,
+                        text=True
+                    )
+                    return result.stdout
+                except subprocess.CalledProcessError as e:
+                    raise RuntimeError(f"Failed to list VACs: {str(e)}")
+                except Exception as e:
+                    raise RuntimeError(f"Error accessing Sunholo: {str(e)}")
+
+            raise ValueError(f"Unknown resource: {uri}")
+
+    def setup_tool_handlers(self):
+        """Configure tool-related handlers"""
+
+        @self.server.list_tools()
+        async def list_tools() -> list[Tool]:
+            """List available Sunholo tools"""
+            return [
+                Tool(
+                    name="chat_with_vac",
+                    description="Chat with a specific Sunholo VAC",
+                    inputSchema={
+                        "type": "object",
+                        "properties": {
+                            "vac_name": {
+                                "type": "string",
+                                "description": "Name of the VAC to chat with"
+                            },
+                            "message": {
+                                "type": "string",
+                                "description": "Message to send to the VAC"
+                            },
+                            "headless": {
+                                "type": "boolean",
+                                "description": "Whether to run in headless mode",
+                                "default": True
+                            }
+                        },
+                        "required": ["vac_name", "message"]
+                    }
+                ),
+                Tool(
+                    name="embed_content",
+                    description="Embed content in a VAC's vector store",
+                    inputSchema={
+                        "type": "object",
+                        "properties": {
+                            "vac_name": {
+                                "type": "string",
+                                "description": "Name of the VAC to embed content for"
+                            },
+                            "content": {
+                                "type": "string",
+                                "description": "Content to embed"
+                            },
+                            "local_chunks": {
+                                "type": "boolean",
+                                "description": "Whether to process chunks locally",
+                                "default": False
+                            }
+                        },
+                        "required": ["vac_name", "content"]
+                    }
+                )
+            ]
+
+        @self.server.call_tool()
+        async def call_tool(
+            name: str,
+            arguments: Any
+        ) -> Sequence[TextContent | ImageContent | EmbeddedResource]:
+            """Handle tool calls for Sunholo interactions"""
+            
+            if name == "chat_with_vac":
+                if not isinstance(arguments, dict):
+                    raise ValueError("Invalid arguments format")
+
+                vac_name = arguments.get("vac_name")
+                message = arguments.get("message")
+                headless = arguments.get("headless", True)
+
+                if not vac_name or not message:
+                    raise ValueError("Missing required arguments")
+
+                try:
+                    cmd = ["sunholo", "vac", "chat", vac_name, message]
+                    if headless:
+                        cmd.append("--headless")
+
+                    result = subprocess.run(
+                        cmd,
+                        capture_output=True,
+                        text=True
+                    )
+
+                    return [
+                        TextContent(
+                            type="text",
+                            text=result.stdout
+                        )
+                    ]
+                except subprocess.CalledProcessError as e:
+                    return [
+                        TextContent(
+                            type="text",
+                            text=f"Error chatting with VAC: {e.stderr}"
+                        )
+                    ]
+
+            elif name == "embed_content":
+                if not isinstance(arguments, dict):
+                    raise ValueError("Invalid arguments format")
+
+                vac_name = arguments.get("vac_name")
+                content = arguments.get("content")
+                local_chunks = arguments.get("local_chunks", False)
+
+                if not vac_name or not content:
+                    raise ValueError("Missing required arguments")
+
+                try:
+                    cmd = ["sunholo", "embed", vac_name, content]
+                    if local_chunks:
+                        cmd.append("--local-chunks")
+
+                    result = subprocess.run(
+                        cmd,
+                        capture_output=True,
+                        text=True
+                    )
+
+                    return [
+                        TextContent(
+                            type="text",
+                            text=result.stdout
+                        )
+                    ]
+                except subprocess.CalledProcessError as e:
+                    return [
+                        TextContent(
+                            type="text",
+                            text=f"Error embedding content: {e.stderr}"
+                        )
+                    ]
+
+            raise ValueError(f"Unknown tool: {name}")
+
+    async def run(self):
+        """Run the MCP server"""
+        async with stdio_server() as (read_stream, write_stream):
+            await self.server.run(
+                read_stream,
+                write_stream,
+                self.server.create_initialization_options()
+            )
+
+def cli_mcp(args):
+    """CLI handler for the MCP server command"""
+    try:
+
+        # Create and run the MCP server
+        server = SunholoMCPServer()
+            
+        logger.info("Starting Sunholo MCP server...")
+        asyncio.run(server.run())
+
+    except Exception as e:
+        logger.error(f"Error running MCP server: {str(e)}")
+        raise
+
+def setup_mcp_subparser(subparsers):
+    """
+    Sets up an argparse subparser for the 'mcp' command.
+
+    By default will use configurations within the folder specified by '_CONFIG_FOLDER'
+
+    Example command:
+    ```bash
+    sunholo mcp
+    ```
+    """
+    mcp_parser = subparsers.add_parser('mcp', 
+                                      help='Start an Anthropic MCP server that wraps `sunholo` functionality')
+    
+    mcp_parser.set_defaults(func=cli_mcp)

{sunholo-0.115.2 → sunholo-0.116.0}/sunholo/utils/config_class.py

@@ -140,7 +140,7 @@ class ConfigManager:
                 config = json.load(file)
             else:
                 # Create YAML parser that forbids duplicates
-                yaml = YAML(typ='full')
+                yaml = YAML(typ='safe')
                 yaml.allow_duplicate_keys = False
                 config = yaml.load(file)
 

{sunholo-0.115.2 → sunholo-0.116.0}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.115.2
+Version: 0.116.0
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.115.2.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.116.0.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0
@@ -138,7 +138,8 @@ Provides-Extra: openai
 Requires-Dist: langchain-openai==0.1.25; extra == "openai"
 Requires-Dist: tiktoken; extra == "openai"
 Provides-Extra: anthropic
-Requires-Dist: langchain-anthropic==0.1.23; extra == "anthropic"
+Requires-Dist: langchain-anthropic>=0.1.23; extra == "anthropic"
+Requires-Dist: mcp; extra == "anthropic"
 Provides-Extra: tools
 Requires-Dist: openapi-spec-validator; extra == "tools"
 Requires-Dist: playwright; extra == "tools"

{sunholo-0.115.2 → sunholo-0.116.0}/sunholo.egg-info/SOURCES.txt

@@ -118,6 +118,8 @@ sunholo/llamaindex/llamaindex_class.py
 sunholo/llamaindex/user_history.py
 sunholo/lookup/__init__.py
 sunholo/lookup/model_lookup.yaml
+sunholo/mcp/__init__.py
+sunholo/mcp/cli.py
 sunholo/pubsub/__init__.py
 sunholo/pubsub/process_pubsub.py
 sunholo/pubsub/pubsub_manager.py

{sunholo-0.115.2 → sunholo-0.116.0}/sunholo.egg-info/requires.txt

@@ -67,7 +67,8 @@ unstructured[all-docs,local-inference]
 xlwings
 
 [anthropic]
-langchain-anthropic==0.1.23
+langchain-anthropic>=0.1.23
+mcp
 
 [azure]
 azure-identity