synapse-framework 0.2.5__tar.gz → 0.2.6__tar.gz

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-framework
-Version: 0.2.5
+Version: 0.2.6
 Summary: Modular Multi-MCP & Hybrid RAG AI Application Framework
 Author: Synapse Contributors
 License-Expression: MIT
@@ -32,9 +32,9 @@ Requires-Dist: langchain-chroma>=1.1.0
 Requires-Dist: langchain-community>=0.4.2
 Requires-Dist: langchain-core>=1.6.7
 Requires-Dist: langchain-ollama>=1.1.0
-Requires-Dist: matplotlib>=3.11.2
+Requires-Dist: matplotlib>=3.8.0
 Requires-Dist: mcp>=2.3.0
-Requires-Dist: networkx>=3.7
+Requires-Dist: networkx>=3.2
 Requires-Dist: pydantic>=2.13.5
 Requires-Dist: uvicorn>=0.30.0
 Provides-Extra: mlflow

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "synapse-framework"
-version = "0.2.5"
+version = "0.2.6"
 description = "Modular Multi-MCP & Hybrid RAG AI Application Framework"
 readme = "README.md"
 license = "MIT"
@@ -50,9 +50,9 @@ dependencies = [
     "langchain-community>=0.4.2",
     "langchain-core>=1.6.7",
     "langchain-ollama>=1.1.0",
-    "matplotlib>=3.11.2",
+    "matplotlib>=3.8.0",
     "mcp>=2.3.0",
-    "networkx>=3.7",
+    "networkx>=3.2",
     "pydantic>=2.13.5",
     "uvicorn>=0.30.0",
 ]

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/synapse/__init__.py

@@ -33,7 +33,7 @@ from synapse import telemetry
 from synapse import telemetry as mlflow
 from synapse.app import BaseApp, SynapseApp
 
-__version__ = "0.2.5"
+__version__ = "0.2.6"
 
 __all__ = [
     "BaseApp",

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/synapse/app.py

@@ -53,9 +53,7 @@ class ChatRequestPayload(BaseModel):
     history: list[ChatMessagePayload] | None = Field(
         default_factory=list, description="Conversation history."
     )
-    use_rag: bool = Field(
-        default=True, description="Enable Hybrid RAG and Multi-MCP tool calls."
-    )
+    use_rag: bool = Field(default=True, description="Enable Hybrid RAG and Multi-MCP tool calls.")
 
 
 class SyncRequestPayload(BaseModel):
@@ -63,7 +61,9 @@ class SyncRequestPayload(BaseModel):
 
     query: str | None = Field(default=None, description="Optional search query filter.")
     max_results: int = Field(default=5, description="Maximum items to index.")
-    extract_graph: bool = Field(default=True, description="Extract semantic knowledge graph triples.")
+    extract_graph: bool = Field(
+        default=True, description="Extract semantic knowledge graph triples."
+    )
     tool_name: str | None = Field(default=None, description="Optional explicit tool name.")
     arguments: dict[str, Any] | None = Field(default=None, description="Optional tool arguments.")
 
@@ -97,9 +97,9 @@ class BaseApp:
         self,
         name: str = "Synapse AI",
         description: str = "Multi-MCP & Hybrid RAG Framework",
-        version: str = "0.2.5",
+        version: str = "0.2.6",
         model: BaseModelProvider | None = None,
-        connectors: list[BaseConnector] | None = None,
+        connectors: list[BaseConnector] | BaseConnector | None = None,
         skills: Skill | Sequence[Skill] | str | Path | None = None,
         resources: Resource | Sequence[Resource] | str | Path | Any | None = None,
         vector_store: VectorStorage | BaseVectorDB | bool | None = None,
@@ -111,6 +111,7 @@ class BaseApp:
         *,
         title: str | None = None,
         app_name: str | None = None,
+        connector: list[BaseConnector] | BaseConnector | None = None,
         resource: Resource | Sequence[Resource] | str | Path | Any | None = None,
         persistent_storage: VectorStorage | BaseVectorDB | bool | None = None,
         Visualize_graph: BaseGraphStore | bool | None = None,
@@ -150,9 +151,7 @@ class BaseApp:
             self.vector_store = VectorStorage(database=storage_arg)
         elif storage_arg is True:
             resolved_data_dir.mkdir(parents=True, exist_ok=True)
-            self.vector_store = VectorStorage(
-                database=chromadb(path=self.data_dir / "chroma_db")
-            )
+            self.vector_store = VectorStorage(database=chromadb(path=self.data_dir / "chroma_db"))
         else:
             self.vector_store = None
 
@@ -179,8 +178,19 @@ class BaseApp:
         )
 
         self._connectors: dict[str, BaseConnector] = {}
-        for conn in connectors or []:
-            self.add_connector(conn)
+        active_connectors = connectors if connectors is not None else connector
+        if active_connectors is not None:
+            if isinstance(active_connectors, BaseConnector):
+                self.add_connector(active_connectors)
+            elif isinstance(active_connectors, (list, tuple, set)):
+                for conn in active_connectors:
+                    self.add_connector(conn)
+            else:
+                try:
+                    for conn in active_connectors:
+                        self.add_connector(conn)
+                except TypeError:
+                    self.add_connector(active_connectors)
 
         self.sync_state: dict[str, Any] = {
             "is_syncing": False,
@@ -193,7 +203,13 @@ class BaseApp:
 
         self._skills: list[Skill] = []
         if skills is not None:
-            if isinstance(skills, (list, tuple)):
+            if callable(skills) and not isinstance(skills, Skill):
+                for sk_name in ("rag-analyst", "knowledge-graph", "mcp-orchestrator"):
+                    try:
+                        self.add_skill(load_premade_skill(sk_name))
+                    except Exception:
+                        pass  # nosec B110
+            elif isinstance(skills, (list, tuple)):
                 for s in skills:
                     self.add_skill(s)
             else:
@@ -269,18 +285,22 @@ class BaseApp:
                 collected.extend(conn._cached_schemas)
             elif hasattr(conn, "tools_dict") and isinstance(conn.tools_dict, dict):
                 for tname in conn.tools_dict.keys():
-                    collected.append({
+                    collected.append(
+                        {
+                            "server": conn.name,
+                            "name": tname,
+                            "connector": conn.name,
+                        }
+                    )
+            elif getattr(conn, "default_tool", None):
+                collected.append(
+                    {
                         "server": conn.name,
-                        "name": tname,
+                        "name": conn.default_tool,
+                        "description": conn.description,
                         "connector": conn.name,
-                    })
-            elif getattr(conn, "default_tool", None):
-                collected.append({
-                    "server": conn.name,
-                    "name": conn.default_tool,
-                    "description": conn.description,
-                    "connector": conn.name,
-                })
+                    }
+                )
         return collected
 
     async def list_tools(self) -> list[dict[str, Any]]:
@@ -296,18 +316,22 @@ class BaseApp:
             if isinstance(tools, list):
                 collected.extend(tools)
             elif getattr(conn, "default_tool", None):
-                collected.append({
-                    "server": conn.name,
-                    "name": conn.default_tool,
-                    "description": conn.description,
-                    "connector": conn.name,
-                })
+                collected.append(
+                    {
+                        "server": conn.name,
+                        "name": conn.default_tool,
+                        "description": conn.description,
+                        "connector": conn.name,
+                    }
+                )
         return collected
 
     def add_skill(self, skill: Skill | str | Path | list[Any] | tuple[Any, ...]) -> Any:
         """Registers a Skill or sequence of skills into this application."""
         if isinstance(skill, (list, tuple)):
             return [self.add_skill(s) for s in skill]
+        if callable(skill) and not isinstance(skill, Skill):
+            return None
         sk = skill if isinstance(skill, Skill) else makeskills(skill)
         self._skills.append(sk)
         return sk
@@ -353,7 +377,6 @@ class BaseApp:
         """Returns all registered resource collections."""
         return self.list_resources()
 
-
     def get_resource(self, name: str) -> Resource | None:
         """Retrieves a resource collection by name."""
         for r in self._resources:
@@ -378,9 +401,11 @@ class BaseApp:
 
         records: list[dict[str, Any]] = []
         if isinstance(raw_result, list):
-            records = [r for r in raw_result if isinstance(r, dict)]
+            records = [r if isinstance(r, dict) else {"result": r} for r in raw_result]
         elif isinstance(raw_result, dict):
             records = [raw_result]
+        elif raw_result is not None:
+            records = [{"result": raw_result}]
 
         formatted_items: list[str] = []
         live_chunks: list[dict[str, Any]] = []
@@ -445,7 +470,9 @@ class BaseApp:
                     self.sync_state.get("items_synced", 0), self.vector_store.count()
                 )
             else:
-                self.sync_state["items_synced"] = self.sync_state.get("items_synced", 0) + len(records)
+                self.sync_state["items_synced"] = self.sync_state.get("items_synced", 0) + len(
+                    records
+                )
         except Exception:
             self.sync_state["items_synced"] = self.sync_state.get("items_synced", 0) + len(records)
 
@@ -476,11 +503,28 @@ class BaseApp:
         """Routes a prompt to appropriate connector tools and executes them."""
         clean_msg = (user_message or "").strip().lower()
         conversational_greetings = {
-            "hi", "hello", "hey", "hola", "greetings", "good morning",
-            "good afternoon", "good evening", "how are you", "who are you",
-            "what can you do", "which model is this", "which model is getting used",
-            "what model is this", "what model is being used", "help", "thanks",
-            "thank you", "bye", "goodbye", "test", "ping",
+            "hi",
+            "hello",
+            "hey",
+            "hola",
+            "greetings",
+            "good morning",
+            "good afternoon",
+            "good evening",
+            "how are you",
+            "who are you",
+            "what can you do",
+            "which model is this",
+            "which model is getting used",
+            "what model is this",
+            "what model is being used",
+            "help",
+            "thanks",
+            "thank you",
+            "bye",
+            "goodbye",
+            "test",
+            "ping",
         }
         if clean_msg in conversational_greetings:
             return {
@@ -512,18 +556,22 @@ class BaseApp:
             for conn, tools in zip(matching_active_conns, tool_lists):
                 if isinstance(tools, list):
                     for t in tools:
-                        candidate_tools.append({
-                            "server": conn.name,
-                            "name": t.get("name", conn.default_tool),
-                            "description": t.get("description", conn.description),
-                            "inputSchema": t.get("inputSchema", {}),
-                        })
+                        candidate_tools.append(
+                            {
+                                "server": conn.name,
+                                "name": t.get("name", conn.default_tool),
+                                "description": t.get("description", conn.description),
+                                "inputSchema": t.get("inputSchema", {}),
+                            }
+                        )
                 if conn.default_tool and conn.intent_regex:
-                    fallback_calls.append({
-                        "server": conn.name,
-                        "tool": conn.default_tool,
-                        "arguments": conn.default_args_builder(user_message),
-                    })
+                    fallback_calls.append(
+                        {
+                            "server": conn.name,
+                            "tool": conn.default_tool,
+                            "arguments": conn.default_args_builder(user_message),
+                        }
+                    )
 
         unauth_notes: list[str] = []
         for c_name in unauthenticated_connectors:
@@ -574,16 +622,16 @@ class BaseApp:
                 context_sections.append(res["section_text"])
                 live_chunks.extend(res["live_chunks"])
                 live_triples.extend(res["live_triples"])
-                executed.append({
-                    "server": srv,
-                    "tool": tl,
-                    "arguments": args,
-                    "items_fetched": len(res["records"]),
-                })
-            except Exception as err:
-                context_sections.append(
-                    f"### {srv.upper()} NOTE:\nLive fetch ({tl}) error: {err}."
+                executed.append(
+                    {
+                        "server": srv,
+                        "tool": tl,
+                        "arguments": args,
+                        "items_fetched": len(res["records"]),
+                    }
                 )
+            except Exception as err:
+                context_sections.append(f"### {srv.upper()} NOTE:\nLive fetch ({tl}) error: {err}.")
 
         return {
             "tool_calls": executed,
@@ -640,9 +688,7 @@ class BaseApp:
             yield {"type": "context", "data": context_data}
 
         context_str = (
-            "\n\n".join(context_sections)
-            if context_sections
-            else "No external context required."
+            "\n\n".join(context_sections) if context_sections else "No external context required."
         )
 
         skills_section = ""
@@ -680,8 +726,12 @@ class BaseApp:
                 response="".join(full_tokens),
                 latency_ms=latency_ms,
                 model=getattr(self.model, "model_name", "custom"),
-                chunks_retrieved=len(merged_chunks) if "merged_chunks" in locals() else len(live_tool_chunks),
-                graph_triples=len(merged_triples) if "merged_triples" in locals() else len(live_tool_triples),
+                chunks_retrieved=len(merged_chunks)
+                if "merged_chunks" in locals()
+                else len(live_tool_chunks),
+                graph_triples=len(merged_triples)
+                if "merged_triples" in locals()
+                else len(live_tool_triples),
                 tools_executed=executed_tools,
             )
 
@@ -735,9 +785,7 @@ class BaseApp:
         async def chat_route(req: ChatRequestPayload):
             async def event_generator():
                 try:
-                    async for ev in app_ref.chat_stream(
-                        req.message, req.history, req.use_rag
-                    ):
+                    async for ev in app_ref.chat_stream(req.message, req.history, req.use_rag):
                         if ev["type"] == "context":
                             yield f"event: context\ndata: {json.dumps(ev['data'])}\n\n"
                         elif ev["type"] == "token":
@@ -785,7 +833,9 @@ class BaseApp:
         ):
             conn = app_ref.get_connector(connector_name)
             if not conn:
-                raise HTTPException(status_code=404, detail=f"Connector '{connector_name}' not found.")
+                raise HTTPException(
+                    status_code=404, detail=f"Connector '{connector_name}' not found."
+                )
 
             tool = req.tool_name or conn.default_tool
             if req.arguments:
@@ -817,7 +867,10 @@ class BaseApp:
                     app_ref.sync_state["is_syncing"] = False
 
             background_tasks.add_task(_bg_task)
-            return {"status": "started", "message": f"Background sync initiated for '{connector_name}'."}
+            return {
+                "status": "started",
+                "message": f"Background sync initiated for '{connector_name}'.",
+            }
 
         # ---------------- Auth Routes ----------------
         @api.get("/auth/status", tags=["Auth"])
@@ -828,7 +881,11 @@ class BaseApp:
                 if hasattr(c, "oauth") and hasattr(c.oauth, "has_credentials"):
                     has_creds = c.oauth.has_credentials()
                 elif hasattr(c, "has_credentials"):
-                    has_creds = c.has_credentials() if callable(c.has_credentials) else bool(c.has_credentials)
+                    has_creds = (
+                        c.has_credentials()
+                        if callable(c.has_credentials)
+                        else bool(c.has_credentials)
+                    )
                 else:
                     has_creds = c.is_authenticated()
 
@@ -992,6 +1049,7 @@ class BaseApp:
                 base_url = getattr(app_ref.model, "base_url", "http://localhost:11434")
                 try:
                     import httpx
+
                     async with httpx.AsyncClient(timeout=0.4) as client:
                         res = await client.get(f"{base_url}/api/tags")
                         model_status = "ready" if res.status_code == 200 else "unreachable"
@@ -1028,7 +1086,11 @@ class BaseApp:
 
             storage_status = "disabled"
             if app_ref.vector_store is not None:
-                storage_status = "ready" if (chroma_path.exists() or hasattr(app_ref.vector_store, "database")) else "initialized"
+                storage_status = (
+                    "ready"
+                    if (chroma_path.exists() or hasattr(app_ref.vector_store, "database"))
+                    else "initialized"
+                )
 
             services_dict = {
                 provider_type: model_status,
@@ -1074,7 +1136,11 @@ class BaseApp:
                 await self.model.close()
             except Exception:
                 pass  # nosec B110
-        if self.graph_store is not None and hasattr(self.graph_store, "driver") and self.graph_store.driver:
+        if (
+            self.graph_store is not None
+            and hasattr(self.graph_store, "driver")
+            and self.graph_store.driver
+        ):
             try:
                 self.graph_store.driver.close()
             except Exception:
@@ -1094,6 +1160,7 @@ class BaseApp:
     def to_mcp_server(self, name: str | None = None) -> Any:
         """Converts all registered connectors into an official MCP Server instance."""
         from synapse.export.mcp_exporter import MCPExporter
+
         return MCPExporter.create_mcp_server(self, name=name or self.name)
 
     def run_mcp(self, transport: str = "stdio") -> None:

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/synapse/export/__init__.py

@@ -5,10 +5,12 @@ from synapse.storage.graph_exporter import GraphExporter
 
 # Developer-friendly alias
 graphexporter = GraphExporter
+mcp_exporter = mcpexporter
 
 __all__ = [
     "GraphExporter",
     "MCPExporter",
     "graphexporter",
+    "mcp_exporter",
     "mcpexporter",
 ]

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/synapse/models/ollama.py

@@ -34,7 +34,7 @@ class OllamaModel(BaseModelProvider):
         temperature: float = DEFAULT_TEMPERATURE,
         num_ctx: int = 4096,
         timeout: float = 60.0,
-        num_predict: int = 256,
+        num_predict: int = 1024,
     ) -> None:
         """Initializes the Ollama model provider.
 
@@ -44,7 +44,7 @@ class OllamaModel(BaseModelProvider):
             temperature: Sampling temperature. Defaults to 0.1.
             num_ctx: Context window size. Defaults to 4096.
             timeout: HTTP request timeout in seconds. Defaults to 60.0.
-            num_predict: Maximum tokens to generate per response turn. Defaults to 256.
+            num_predict: Maximum tokens to generate per response turn. Defaults to 1024.
         """
         super().__init__(model_name=model, temperature=temperature)
         self.base_url = base_url.rstrip("/")
@@ -116,6 +116,7 @@ class OllamaModel(BaseModelProvider):
         prompt: str,
         temperature: float | None = None,
         format: str | None = None,
+        num_predict: int | None = None,
     ) -> str:
         """Generates a complete non-streaming response from Ollama.
 
@@ -123,11 +124,13 @@ class OllamaModel(BaseModelProvider):
             prompt: Text prompt string.
             temperature: Optional temperature override.
             format: Optional output format (e.g. 'json').
+            num_predict: Optional token generation limit override.
 
         Returns:
             The complete response text string.
         """
         temp = self.temperature if temperature is None else temperature
+        pred = num_predict if num_predict is not None else self.num_predict
         payload: dict[str, Any] = {
             "model": self.model_name,
             "prompt": prompt,
@@ -135,7 +138,7 @@ class OllamaModel(BaseModelProvider):
             "options": {
                 "temperature": temp,
                 "num_ctx": self.num_ctx,
-                "num_predict": self.num_predict,
+                "num_predict": pred,
             },
         }
         if format:
@@ -147,9 +150,20 @@ class OllamaModel(BaseModelProvider):
                     f"{self.base_url}/api/generate",
                     json=payload,
                 )
+                if res.status_code == 501 and format:
+                    # MLX runner or models that do not support format: json
+                    payload.pop("format", None)
+                    res = await client.post(
+                        f"{self.base_url}/api/generate",
+                        json=payload,
+                    )
                 res.raise_for_status()
                 data = res.json()
-                return str(data.get("response", "")).strip()
+                resp_text = str(data.get("response", "")).strip()
+                if not resp_text and data.get("thinking"):
+                    # Fallback for reasoning/thinking models if response is empty
+                    resp_text = str(data.get("thinking", "")).strip()
+                return resp_text
         except (httpx.ConnectError, httpx.ConnectTimeout):
             return f"[Error: Cannot reach Ollama at {self.base_url}. Ensure Ollama is running.]"
         except Exception as e:
@@ -193,7 +207,15 @@ class OllamaModel(BaseModelProvider):
                 cleaned = match.group(0)
             parsed = json.loads(cleaned)
             if isinstance(parsed, dict):
-                parsed = parsed.get("triples") or parsed.get("data") or (list(parsed.values())[0] if parsed.values() and isinstance(list(parsed.values())[0], list) else [])
+                parsed = (
+                    parsed.get("triples")
+                    or parsed.get("data")
+                    or (
+                        list(parsed.values())[0]
+                        if parsed.values() and isinstance(list(parsed.values())[0], list)
+                        else []
+                    )
+                )
 
             if isinstance(parsed, list):
                 valid_triples: list[dict[str, str]] = []
@@ -204,26 +226,21 @@ class OllamaModel(BaseModelProvider):
                         and "relation" in item
                         and "target" in item
                     ):
-                        valid_triples.append({
-                            "source": str(item["source"]).strip(),
-                            "relation": (
-                                str(item["relation"])
-                                .strip()
-                                .upper()
-                                .replace(" ", "_")
-                            ),
-                            "target": str(item["target"]).strip(),
-                            "source_type": (
-                                str(item.get("source_type", "Default"))
-                                .strip()
-                                .title()
-                            ),
-                            "target_type": (
-                                str(item.get("target_type", "Default"))
-                                .strip()
-                                .title()
-                            ),
-                        })
+                        valid_triples.append(
+                            {
+                                "source": str(item["source"]).strip(),
+                                "relation": (
+                                    str(item["relation"]).strip().upper().replace(" ", "_")
+                                ),
+                                "target": str(item["target"]).strip(),
+                                "source_type": (
+                                    str(item.get("source_type", "Default")).strip().title()
+                                ),
+                                "target_type": (
+                                    str(item.get("target_type", "Default")).strip().title()
+                                ),
+                            }
+                        )
                 return valid_triples
         except Exception:
             return []
@@ -266,6 +283,73 @@ class OllamaModel(BaseModelProvider):
             valid_keys[("", name.lower())] = t
             valid_keys[(srv.lower(), f"{srv.lower()}_{name.lower()}")] = t
 
+        # 1. Primary path: Attempt native Ollama function/tool calling via /api/chat
+        try:
+            formatted_tools: list[dict[str, Any]] = []
+            for t in candidate_tools:
+                t_name = str(t.get("name") or "").strip()
+                t_desc = str(t.get("description") or "").strip()
+                t_params = (
+                    t.get("inputSchema")
+                    or t.get("parameters")
+                    or {"type": "object", "properties": {}}
+                )
+                formatted_tools.append(
+                    {
+                        "type": "function",
+                        "function": {
+                            "name": t_name,
+                            "description": t_desc,
+                            "parameters": t_params,
+                        },
+                    }
+                )
+
+            async with httpx.AsyncClient(timeout=self.timeout) as client:
+                chat_res = await client.post(
+                    f"{self.base_url}/api/chat",
+                    json={
+                        "model": self.model_name,
+                        "messages": [{"role": "user", "content": text}],
+                        "tools": formatted_tools,
+                        "stream": False,
+                        "options": {
+                            "temperature": 0.0,
+                        },
+                    },
+                )
+                if chat_res.status_code == 200:
+                    chat_data = chat_res.json()
+                    native_calls = chat_data.get("message", {}).get("tool_calls") or []
+                    if isinstance(native_calls, list) and len(native_calls) > 0:
+                        validated_native: list[dict[str, Any]] = []
+                        for call in native_calls:
+                            fn = call.get("function") or {}
+                            tl_name = str(fn.get("name") or "").strip()
+                            args = fn.get("arguments")
+                            if not isinstance(args, dict):
+                                args = {}
+                            spec = valid_keys.get(("", tl_name)) or valid_keys.get(
+                                ("", tl_name.lower())
+                            )
+                            if not spec:
+                                for (_k_srv, k_name), s_val in valid_keys.items():
+                                    if k_name.lower() == tl_name.lower():
+                                        spec = s_val
+                                        break
+                            if spec:
+                                validated_native.append(
+                                    {
+                                        "server": str(spec.get("server") or ""),
+                                        "tool": str(spec.get("name") or tl_name),
+                                        "arguments": args,
+                                    }
+                                )
+                        if validated_native:
+                            return validated_native
+        except Exception:
+            pass  # nosec B110
+
         catalog_text = "\n".join(catalog_lines)
         tool_prompt = (
             "You are a multi-tool router for a personal AI assistant.\n"
@@ -283,7 +367,12 @@ class OllamaModel(BaseModelProvider):
         )
 
         try:
-            raw_text = await self.generate(tool_prompt, temperature=0.0, format="json")
+            raw_text = await self.generate(
+                tool_prompt,
+                temperature=0.0,
+                format="json",
+                num_predict=max(self.num_predict, 1024),
+            )
             if not raw_text or raw_text.startswith("[Error:"):
                 return []
             cleaned = re.sub(
@@ -292,7 +381,9 @@ class OllamaModel(BaseModelProvider):
                 raw_text,
                 flags=re.MULTILINE,
             ).strip()
-            match = re.search(r"\{.*\}", cleaned, re.DOTALL)
+            match = re.search(r"\{\s*\"tool_calls\".*\}", cleaned, re.DOTALL) or re.search(
+                r"\{.*\}", cleaned, re.DOTALL
+            )
             if match:
                 cleaned = match.group(0)
             parsed = json.loads(cleaned)
@@ -323,11 +414,13 @@ class OllamaModel(BaseModelProvider):
                         args = item.get("arguments")
                         if not isinstance(args, dict):
                             args = {}
-                        validated.append({
-                            "server": resolved_srv,
-                            "tool": resolved_tl,
-                            "arguments": args,
-                        })
+                        validated.append(
+                            {
+                                "server": resolved_srv,
+                                "tool": resolved_tl,
+                                "arguments": args,
+                            }
+                        )
                     return validated
         except (json.JSONDecodeError, KeyError, TypeError, ValueError, httpx.HTTPError):
             pass

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/synapse_framework.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-framework
-Version: 0.2.5
+Version: 0.2.6
 Summary: Modular Multi-MCP & Hybrid RAG AI Application Framework
 Author: Synapse Contributors
 License-Expression: MIT
@@ -32,9 +32,9 @@ Requires-Dist: langchain-chroma>=1.1.0
 Requires-Dist: langchain-community>=0.4.2
 Requires-Dist: langchain-core>=1.6.7
 Requires-Dist: langchain-ollama>=1.1.0
-Requires-Dist: matplotlib>=3.11.2
+Requires-Dist: matplotlib>=3.8.0
 Requires-Dist: mcp>=2.3.0
-Requires-Dist: networkx>=3.7
+Requires-Dist: networkx>=3.2
 Requires-Dist: pydantic>=2.13.5
 Requires-Dist: uvicorn>=0.30.0
 Provides-Extra: mlflow

{synapse_framework-0.2.5 → synapse_framework-0.2.6}/synapse_framework.egg-info/requires.txt

@@ -10,9 +10,9 @@ langchain-chroma>=1.1.0
 langchain-community>=0.4.2
 langchain-core>=1.6.7
 langchain-ollama>=1.1.0
-matplotlib>=3.11.2
+matplotlib>=3.8.0
 mcp>=2.3.0
-networkx>=3.7
+networkx>=3.2
 pydantic>=2.13.5
 uvicorn>=0.30.0