fid-mcp 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

fid_mcp/server.py

@@ -204,14 +204,38 @@ class DynamicToolServer:
         }
         self.tools: Dict[str, Dict[str, Any]] = {}
 
+        # Fid knowledge base configuration
+        self.fid_config = None
+        self.fid_pat = os.environ.get("FID_PAT")
+
         # Setup handlers
         self.setup_handlers()
 
     def setup_handlers(self):
         @self.server.list_tools()
         async def handle_list_tools() -> list[types.Tool]:
-            """Return all dynamically loaded tools"""
+            """Return all dynamically loaded tools plus Fid search"""
             tools = []
+
+            # Always add Fid search tool (will show error if not configured when called)
+            tools.append(
+                types.Tool(
+                    name="search",
+                    description="Search the user's Fid knowledge base, which typically contains datasheets, integration manuals, API references, and other technical documentation, particularly for hardware components. ALWAYS try searching Fid first if the user asks about a hardware component, datasheet, manual, or API spec.",
+                    inputSchema={
+                        "type": "object",
+                        "properties": {
+                            "query": {
+                                "type": "string",
+                                "description": "The search query string",
+                            }
+                        },
+                        "required": ["query"],
+                    },
+                )
+            )
+
+            # Add dynamically loaded tools
             for tool_name, tool_def in self.tools.items():
                 # Build input schema from parameters
                 properties = {}
@@ -249,7 +273,33 @@ class DynamicToolServer:
         async def handle_call_tool(
             name: str, arguments: Dict[str, Any] | None
         ) -> list[types.TextContent | types.ImageContent | types.EmbeddedResource]:
-            """Execute a dynamically loaded tool"""
+            """Execute a dynamically loaded tool or Fid search"""
+
+            # Handle Fid search tool
+            if name == "search":
+                if not self.fid_pat:
+                    error_msg = "Personal Access Token (PAT) missing. Please install the Fid MCP server according to these instructions: https://docs.fidlabs.ai/en/latest/connecting-agents.html"
+                    return [types.TextContent(type="text", text=f"Error: {error_msg}")]
+
+                if not self.fid_config:
+                    error_msg = "No Fid toolkit found in current directory. Download an existing toolkit from your Fid project, or create a new one with these instructions: https://docs.fidlabs.ai/en/latest/toolkits.html"
+                    return [types.TextContent(type="text", text=f"Error: {error_msg}")]
+
+                if not arguments or "query" not in arguments:
+                    raise ValueError("Query parameter is required")
+
+                try:
+                    results = await self._search_fid_knowledge(arguments["query"])
+                    return [
+                        types.TextContent(
+                            type="text", text=json.dumps(results, indent=2)
+                        )
+                    ]
+                except Exception as e:
+                    error_msg = f"Search failed: {str(e)}"
+                    return [types.TextContent(type="text", text=f"Error: {error_msg}")]
+
+            # Handle dynamic tools
             if name not in self.tools:
                 raise ValueError(f"Unknown tool: {name}")
 
@@ -324,6 +374,55 @@ class DynamicToolServer:
             output = self._build_output(tool_def.get("output"), exec_context)
             return [types.TextContent(type="text", text=json.dumps(output, indent=2))]
 
+    async def _search_fid_knowledge(self, query: str) -> Dict[str, Any]:
+        """Search the Fid knowledge base"""
+        import aiohttp
+        import time
+
+        start_time = time.time()
+        default_k = 6
+
+        # Use the correct projects/{projectId}/search endpoint
+        url = f"{self.fid_config['apiBaseUrl']}/projects/{self.fid_config['projectId']}/search"
+        headers = {
+            "X-API-Key": self.fid_pat,
+            "X-Client-Source": "mcp_client",
+        }
+
+        payload = {
+            "query": query,
+            "k": default_k,
+            "userId": "",  # Required field, can be empty
+            "search_method": "multi_stage",
+            "multi_stage_methods": ["full_text", "vector"],
+        }
+
+        async with aiohttp.ClientSession() as session:
+            async with session.post(url, json=payload, headers=headers) as response:
+                if response.status != 200:
+                    error_text = await response.text()
+                    raise ValueError(
+                        f"API request failed with status {response.status}: {error_text}"
+                    )
+
+                data = await response.json()
+
+                # Log search time
+                duration = time.time() - start_time
+
+                if not data.get("results"):
+                    return {
+                        "results": [],
+                        "message": "No search results found",
+                        "duration": duration,
+                    }
+
+                return {
+                    "results": data.get("results", []),
+                    "totalCount": data.get("totalCount"),
+                    "duration": duration,
+                }
+
     def load_toolset(self, filepath: Path):
         """Load and register tools from a .fidtools or fidtools.json file"""
         with open(filepath, "r") as f:
@@ -335,6 +434,13 @@ class DynamicToolServer:
         except ValueError as e:
             raise ValueError(f"Failed to validate toolset '{filepath}': {e}")
 
+        # Extract Fid configuration if present
+        if "projectId" in toolset and "apiBaseUrl" in toolset:
+            self.fid_config = {
+                "projectId": toolset["projectId"],
+                "apiBaseUrl": toolset["apiBaseUrl"],
+            }
+
         # Register each tool
         for tool_def in toolset["tools"]:
             self.tools[tool_def["name"]] = tool_def
@@ -426,19 +532,18 @@ class DynamicToolServer:
             f.write(f"Using working_dir: {working_dir}\n")
 
         # Look for .fidtools or fidtools.json file in actual working directory
-        config_files = [
-            working_dir / ".fidtools",
-            working_dir / "fidtools.json"
-        ]
-        
+        config_files = [working_dir / ".fidtools", working_dir / "fidtools.json"]
+
         config_file = None
         for file_path in config_files:
             if file_path.exists() and file_path.is_file():
                 config_file = file_path
                 break
-        
+
         with open(debug_log, "a") as f:
-            f.write(f"Looking for config files: {', '.join(str(f) for f in config_files)}\n")
+            f.write(
+                f"Looking for config files: {', '.join(str(f) for f in config_files)}\n"
+            )
             if config_file:
                 f.write(f"Found config file: {config_file}\n")
             else:

{fid_mcp-0.1.1.dist-info → fid_mcp-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fid-mcp
-Version: 0.1.1
+Version: 0.1.2
 Summary: MCP Server for serving configurable Fid toolkits
 Project-URL: Homepage, https://fidlabs.ai
 Author-email: Fid Labs <contact@fidlabs.ai>
@@ -10,6 +10,7 @@ Keywords: mcp,model-context-protocol,server,tools
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.8
+Requires-Dist: aiohttp
 Requires-Dist: mcp
 Requires-Dist: pexpect
 Requires-Dist: pydantic
@@ -17,12 +18,12 @@ Description-Content-Type: text/markdown
 
 # Fid MCP Server
 
-MCP server that dynamically serves tools according to `.fidtools` configs present in the working directory. `.fidtools` are generated by the Fid web application and intended to allow developers to quickly onboard coding agents to their unique development environments.
+MCP server that dynamically serves tools according to `fidtools.json` or `.fidtools` configs present in the working directory. Toolkits are generated by the Fid web application and intended to allow developers to quickly onboard coding agents to their unique development environments.
 
 To install:
 ```bash
-claude mcp add fid --scope user --env FID_PAT=YOUR_FID_PAT -- uvx fid-mcp
+claude mcp add Fid --scope user --env FID_PAT=YOUR_FID_PAT -- uvx fid-mcp
 ```
 Note: retrieve your PAT from the user menu in your Fid project, or just use the pre-constructed install command provided by the "Connect to your Agent" button.
 
-See [fidlabs.ai](https://fidlabs.ai) for more information or reach out to us at [contact@fidlabs.ai](mailto:contact@fidlabs.ai).
+See [docs.fidlabs.ai](https://docs.fidlabs.ai) for more information or reach out to us at [contact@fidlabs.ai](mailto:contact@fidlabs.ai).

fid_mcp-0.1.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+fid_mcp/__init__.py,sha256=cv_5UWE6kUBeTfAY6JZq9o47hEfzaLhcoKW0rw8GKq8,17
+fid_mcp/config.py,sha256=YEH9BD4UWU9Hp3ifc3PGg4nk3FvFadfRt52-_fhCaqk,7365
+fid_mcp/server.py,sha256=LctNUIjsBzvza6-NGNOxcByHjyFG83ojfxUdoELX8ak,22501
+fid_mcp/shell.py,sha256=mLbGIVuk4LGCT37tS8deDOyTSY2zPG57WE9KBGsKSHg,26459
+fid_mcp-0.1.2.dist-info/METADATA,sha256=VCqD3CDKW5tUkwe-HCtQGLFKFn_trrjxL0m4hBiArrM,1259
+fid_mcp-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fid_mcp-0.1.2.dist-info/entry_points.txt,sha256=Xdh2RDDC7r5rIXe1kM3smsA1rgNxd1tafL_JuuTJf2o,48
+fid_mcp-0.1.2.dist-info/licenses/LICENSE,sha256=5SkiXC7s2RbMDCdmq4N6MCWNKw151sFmkGbWS9BLSJE,177
+fid_mcp-0.1.2.dist-info/RECORD,,

fid_mcp-0.1.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-fid_mcp/__init__.py,sha256=cv_5UWE6kUBeTfAY6JZq9o47hEfzaLhcoKW0rw8GKq8,17
-fid_mcp/config.py,sha256=YEH9BD4UWU9Hp3ifc3PGg4nk3FvFadfRt52-_fhCaqk,7365
-fid_mcp/server.py,sha256=ZlQqNW_b6vfGzjcxxgKPKYy_YNw7FQNZfbjKV0ywA04,18016
-fid_mcp/shell.py,sha256=mLbGIVuk4LGCT37tS8deDOyTSY2zPG57WE9KBGsKSHg,26459
-fid_mcp-0.1.1.dist-info/METADATA,sha256=l348ByqdrsTS43EchBzAFoyqQAj_GtRCUZ0_pd2wKfg,1210
-fid_mcp-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fid_mcp-0.1.1.dist-info/entry_points.txt,sha256=Xdh2RDDC7r5rIXe1kM3smsA1rgNxd1tafL_JuuTJf2o,48
-fid_mcp-0.1.1.dist-info/licenses/LICENSE,sha256=5SkiXC7s2RbMDCdmq4N6MCWNKw151sFmkGbWS9BLSJE,177
-fid_mcp-0.1.1.dist-info/RECORD,,