Fix dos

Author: plutasnyy

uns_mcp/connectors/destination/weaviate.py

@@ -42,9 +42,11 @@ async def create_weaviate_destination(
     Args:
         cluster_url: URL of the weaviate cluster
         collection : Name of the collection to use in the weaviate cluster
-        Note: The collection is a table in the weaviate cluster.
-              In platform, there are dedicated code to generate collection for users
-              here, due to the simplicity of the server, we are not generating it for users.
+
+    Note: The collection is a table in the Weaviate cluster. In the platform, the collection
+        name can be generated automatically; however, this is not yet supported via the API.
+        Therefore, the collection name is required. Please note that the schema cannot be empty
+        and must contain at least a record_id: Text.
 
     Returns:
         String containing the created destination connector information

uns_mcp/server.py

@@ -104,7 +104,7 @@ async def app_lifespan(server: FastMCP) -> AsyncIterator[AppContext]:
 
 
 @mcp.tool()
-async def list_sources(ctx: Context, source_type: Optional[str] = None) -> str:
+async def list_sources(ctx: Context, source_type: Optional[SourceConnectorType] = None) -> str:
     """
     List available sources from the Unstructured API.
 
@@ -118,9 +118,8 @@ async def list_sources(ctx: Context, source_type: Optional[str] = None) -> str:
 
     request = ListSourcesRequest()
     if source_type:
-        source_type = source_type.upper()  # it needs uppercase to access
         try:
-            request.source_type = SourceConnectorType[source_type]
+            request.source_type = SourceConnectorType(source_type)
         except KeyError:
             return f"Invalid source type: {source_type}"
 
@@ -166,7 +165,10 @@ async def get_source_info(ctx: Context, source_id: str) -> str:
 
 
 @mcp.tool()
-async def list_destinations(ctx: Context, destination_type: Optional[str] = None) -> str:
+async def list_destinations(
+    ctx: Context,
+    destination_type: Optional[DestinationConnectorType] = None,
+) -> str:
     """List available destinations from the Unstructured API.
 
     Args:
@@ -179,9 +181,8 @@ async def list_destinations(ctx: Context, destination_type: Optional[str] = None
 
     request = ListDestinationsRequest()
     if destination_type:
-        destination_type = destination_type.upper()
         try:
-            request.destination_type = DestinationConnectorType[destination_type]
+            request.destination_type = DestinationConnectorType(destination_type)
         except KeyError:
             return f"Invalid destination type: {destination_type}"
 
@@ -234,7 +235,7 @@ async def list_workflows(
     ctx: Context,
     destination_id: Optional[str] = None,
     source_id: Optional[str] = None,
-    status: Optional[str] = None,
+    status: Optional[WorkflowState] = None,
 ) -> str:
     """
     List workflows from the Unstructured API.
@@ -253,7 +254,7 @@ async def list_workflows(
 
     if status:
         try:
-            request.status = WorkflowState[status]
+            request.status = WorkflowState(status)
         except KeyError:
             return f"Invalid workflow status: {status}"
 
@@ -442,7 +443,7 @@ async def delete_workflow(ctx: Context, workflow_id: str) -> str:
 async def list_jobs(
     ctx: Context,
     workflow_id: Optional[str] = None,
-    status: Optional[str] = None,
+    status: Optional[JobStatus] = None,
 ) -> str:
     """
     List jobs via the Unstructured API.
@@ -460,7 +461,7 @@ async def list_jobs(
 
     if status:
         try:
-            request.status = JobStatus[status]
+            request.status = JobStatus(status)
         except KeyError:
             return f"Invalid job status: {status}"