.

Author: arthurbrenno

pyproject.toml

@@ -4,7 +4,14 @@ version = "v0.1.4"
 description = "Turn any webpage into structured outputs!"
 readme = "README.md"
 requires-python = ">=3.13"
-dependencies = ["architecture>=0.5.44", "intellibricks>=0.7.24", "markdownify>=0.14.1", "msgspec>=0.19.0", "playwright>=1.49.1", "requests>=2.32.3"]
+dependencies = [
+    "architecture>=0.5.44",
+ "intellibricks>=0.7.24",
+ "markdownify>=0.14.1",
+ "msgspec>=0.19.0",
+ "playwright>=1.49.1",
+ "requests>=2.32.3",
+]
 
 [[project.authors]]
 name = "Arthur Brenno"

src/webson/_webson.py

@@ -37,7 +37,7 @@
 import msgspec
 from architecture import log
 from architecture.utils.functions import run_sync
-from intellibricks.llms import Synapse, SynapseCascade
+from intellibricks.llms import Synapse, SynapseCascade, SynapseProtocol
 from intellibricks.llms.util import get_struct_from_schema
 from markdownify import markdownify as md
 from playwright.async_api import async_playwright, Browser
@@ -59,28 +59,28 @@ class Webson(msgspec.Struct):
     A class for extracting webpage contents and converting them into structured data using an LLM.
 
     Attributes:
-        llm (Synapse | SynapseCascade):
+        llm (SynapseProtocol):
             The underlying LLM interface used for generating structured outputs.
         timeout (int | None):
             Optional timeout (in milliseconds) for page loads in Playwright.
     """
 
-    llm: Synapse | SynapseCascade
+    llm: SynapseProtocol
     """The synapse weaver used to perform its operations."""
 
     timeout: int | None = None
     """Timeout (in milliseconds) to use for page loads. If None, the default timeout is used."""
 
     @classmethod
-    def create(cls, llm: Synapse | SynapseCascade | OpenAI | None) -> Webson:
+    def create(cls, llm: SynapseProtocol | OpenAI | None) -> Webson:
         """
         Creates a Webson instance from an LLM instance.
 
         This factory method accepts an instance of either Synapse, SynapseCascade, or OpenAI.
         When an OpenAI instance is provided, it wraps it as a Synapse using the 'gpt-4o' model.
 
         Args:
-            llm (Synapse | SynapseCascade | OpenAI): The LLM instance to be used.
+            llm (SynapseProtocol | OpenAI): The LLM instance to be used.
 
         Returns:
             Webson: A new instance of Webson configured with the appropriate llm.