Optimise in-line documentation for documentation generation

README.md

@@ -425,12 +425,12 @@ result = agent.get("What's in this image?", "screenshot.png")
 
 #### Using response schemas
 
-For structured data extraction, use Pydantic models extending `JsonSchemaBase`:
+For structured data extraction, use Pydantic models extending `ResponseSchemaBase`:
 
 ```python
-from askui import JsonSchemaBase
+from askui import ResponseSchemaBase
 
-class UserInfo(JsonSchemaBase):
+class UserInfo(ResponseSchemaBase):
     username: str
     is_online: bool
 

src/askui/__init__.py

@@ -3,18 +3,19 @@
 __version__ = "0.3.0"
 
 from .agent import VisionAgent
-from .models.router import ModelRouter
+from .models import ModelComposition, ModelDefinition
+from .models.router import Point
 from .models.types.response_schemas import ResponseSchema, ResponseSchemaBase
-from .tools.toolbox import AgentToolbox
-from .tools.agent_os import AgentOs, ModifierKey, PcKey
-
+from .tools import ModifierKey, PcKey
+from .utils.image_utils import Img
 
 __all__ = [
-    "AgentOs",
-    "AgentToolbox",
-    "ModelRouter",
+    "Img",
+    "ModelComposition",
+    "ModelDefinition",
     "ModifierKey",
     "PcKey",
+    "Point",
     "ResponseSchema",
     "ResponseSchemaBase",
     "VisionAgent",

src/askui/locators/__init__.py

@@ -1,9 +1,16 @@
-from askui.locators.locators import AiElement, Element, Prompt, Image, Text
+from .locators import AiElement, Element, Prompt, Image, Text, TextMatchType, Locator
+from .relatable import CircularDependencyError, ReferencePoint, RelationIndex, Relatable
 
 __all__ = [
     "AiElement",
+    "CircularDependencyError",
     "Element",
-    "Prompt",
     "Image",
+    "Locator",
+    "Prompt",
+    "ReferencePoint",
+    "Relatable",
+    "RelationIndex",
     "Text",
+    "TextMatchType",
 ]