struai 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

struai/__init__.py

@@ -60,6 +60,8 @@ from .models import (
     QueryResponse,
     # Entities
     Entity,
+    EntityListItem,
+    EntityRelation,
     Fact,
 )
 
@@ -104,5 +106,7 @@ __all__ = [
     "SearchHit",
     "QueryResponse",
     "Entity",
+    "EntityListItem",
+    "EntityRelation",
     "Fact",
 ]

struai/_base.py

@@ -1,6 +1,7 @@
 """Base HTTP client with retry logic."""
 import time
 from typing import Any, Dict, Optional, Type, TypeVar, Union
+from urllib.parse import urlparse
 
 import httpx
 from pydantic import BaseModel
@@ -20,11 +21,24 @@ from ._version import __version__
 
 T = TypeVar("T", bound=BaseModel)
 
-DEFAULT_BASE_URL = "https://api.stru.ai/v1"
+DEFAULT_BASE_URL = "https://api.stru.ai"
 DEFAULT_TIMEOUT = 60.0
 DEFAULT_MAX_RETRIES = 2
 
 
+def _normalize_base_url(base_url: str) -> str:
+    trimmed = base_url.rstrip("/")
+    parsed = urlparse(trimmed)
+    if parsed.scheme and parsed.netloc:
+        path = parsed.path.rstrip("/")
+        if path in ("", "/"):
+            return f"{trimmed}/v1"
+        return trimmed
+    if trimmed.endswith("/v1"):
+        return trimmed
+    return f"{trimmed}/v1"
+
+
 class BaseClient:
     """Base HTTP client with retry logic."""
 
@@ -36,7 +50,7 @@ class BaseClient:
         max_retries: int = DEFAULT_MAX_RETRIES,
     ):
         self.api_key = api_key
-        self.base_url = base_url.rstrip("/")
+        self.base_url = _normalize_base_url(base_url)
         self.timeout = timeout
         self.max_retries = max_retries
         self._client: Optional[httpx.Client] = None
@@ -195,7 +209,7 @@ class AsyncBaseClient:
         max_retries: int = DEFAULT_MAX_RETRIES,
     ):
         self.api_key = api_key
-        self.base_url = base_url.rstrip("/")
+        self.base_url = _normalize_base_url(base_url)
         self.timeout = timeout
         self.max_retries = max_retries
         self._client: Optional[httpx.AsyncClient] = None

struai/_client.py

@@ -14,7 +14,8 @@ class StruAI(BaseClient):
 
     Args:
         api_key: Your API key. Falls back to STRUAI_API_KEY env var.
-        base_url: API base URL. Defaults to windowseat internal endpoint.
+        base_url: API base URL. Defaults to https://api.stru.ai.
+            If no path is provided, /v1 is appended automatically.
         timeout: Request timeout in seconds. Default 60.
         max_retries: Max retry attempts for failed requests. Default 2.
 

struai/_version.py

@@ -1,2 +1,2 @@
 """Version information."""
-__version__ = "0.1.0"
+__version__ = "0.2.0"

struai/models/__init__.py

@@ -10,7 +10,7 @@ from .drawings import (
     SectionTag,
     TitleBlock,
 )
-from .entities import Entity, EntityLocation, Fact
+from .entities import Entity, EntityListItem, EntityLocation, EntityRelation, Fact
 from .projects import JobStatus, JobStep, Project, Sheet, SheetResult
 from .search import (
     EntitySummary,
@@ -55,6 +55,8 @@ __all__ = [
     "QuerySource",
     # Entities
     "Entity",
+    "EntityListItem",
     "EntityLocation",
+    "EntityRelation",
     "Fact",
 ]

struai/models/entities.py

@@ -3,27 +3,53 @@ from typing import List, Optional
 
 from pydantic import BaseModel
 
+from .common import BBox
+
 
 class EntityLocation(BaseModel):
     """Where an entity appears."""
 
     sheet_id: str
     sheet_title: Optional[str] = None
-    page: int
+    page: Optional[int] = None
+
+
+class EntityRelation(BaseModel):
+    """Relationship entry on an entity detail response."""
+
+    uuid: str
+    type: str
+    fact: str
+    source_id: Optional[str] = None
+    source_label: Optional[str] = None
+    target_id: Optional[str] = None
+    target_label: Optional[str] = None
 
 
 class Fact(BaseModel):
-    """Relationship between entities."""
+    """Relationship between entities (list endpoint)."""
 
     id: str
+    type: str
     fact: str
-    edge_type: str
     source_id: str
-    source_label: Optional[str] = None
     target_id: str
+    source_label: Optional[str] = None
     target_label: Optional[str] = None
 
 
+class EntityListItem(BaseModel):
+    """Entity summary from list endpoint."""
+
+    id: str
+    type: str
+    label: str
+    description: Optional[str] = None
+    sheet_id: Optional[str] = None
+    bbox: Optional[BBox] = None
+    attributes: Optional[str] = None
+
+
 class Entity(BaseModel):
     """Full entity with relationships."""
 
@@ -31,7 +57,6 @@ class Entity(BaseModel):
     type: str
     label: str
     description: Optional[str] = None
-    group_id: str
-    outgoing_facts: List[Fact] = []
-    incoming_facts: List[Fact] = []
+    outgoing: List[EntityRelation] = []
+    incoming: List[EntityRelation] = []
     locations: List[EntityLocation] = []

struai/resources/projects.py

@@ -6,7 +6,7 @@ from pathlib import Path
 from typing import TYPE_CHECKING, BinaryIO, Dict, List, Optional, Union
 
 from .._exceptions import JobFailedError, TimeoutError
-from ..models.entities import Entity, Fact
+from ..models.entities import Entity, EntityListItem, Fact
 from ..models.projects import JobStatus, Project, Sheet, SheetResult
 from ..models.search import QueryResponse, SearchResponse
 
@@ -266,7 +266,7 @@ class Entities:
         sheet_id: Optional[str] = None,
         type: Optional[str] = None,
         limit: int = 100,
-    ) -> List[Entity]:
+    ) -> List[EntityListItem]:
         """List entities in project."""
         params: Dict[str, Union[str, int]] = {"limit": limit}
         if sheet_id:
@@ -278,7 +278,7 @@ class Entities:
             f"/projects/{self._project_id}/entities",
             params=params,
         )
-        return [Entity.model_validate(e) for e in response["entities"]]
+        return [EntityListItem.model_validate(e) for e in response["entities"]]
 
     def get(self, entity_id: str) -> Entity:
         """Get entity with all relationships."""
@@ -300,7 +300,7 @@ class AsyncEntities:
         sheet_id: Optional[str] = None,
         type: Optional[str] = None,
         limit: int = 100,
-    ) -> List[Entity]:
+    ) -> List[EntityListItem]:
         """List entities in project."""
         params: Dict[str, Union[str, int]] = {"limit": limit}
         if sheet_id:
@@ -312,7 +312,7 @@ class AsyncEntities:
             f"/projects/{self._project_id}/entities",
             params=params,
         )
-        return [Entity.model_validate(e) for e in response["entities"]]
+        return [EntityListItem.model_validate(e) for e in response["entities"]]
 
     async def get(self, entity_id: str) -> Entity:
         """Get entity with all relationships."""

{struai-0.1.0.dist-info → struai-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: struai
-Version: 0.1.0
+Version: 0.2.0
 Summary: StruAI Drawing Analysis SDK - AI-powered construction drawing analysis
 Project-URL: Homepage, https://struai.com
 Project-URL: Documentation, https://docs.struai.com/python
@@ -46,6 +46,9 @@ pip install struai
 from struai import StruAI
 
 client = StruAI(api_key="sk-xxx")  # or set STRUAI_API_KEY env var
+
+# Optional: override base URL (http://localhost:8000 or http://localhost:8000/v1)
+client = StruAI(api_key="sk-xxx", base_url="http://localhost:8000")
 ```
 
 ## Tier 1: Raw Detection ($0.02/page)

struai-0.2.0.dist-info/RECORD

@@ -0,0 +1,18 @@
+struai/__init__.py,sha256=SmHeLcTTzSUNIYXmiDmE2_LRxZl8fUYQwhHHCvJwoGY,2274
+struai/_base.py,sha256=JiAxWI1PMmIFVr9v_hBABra8m2fRkEvBp7d2Vc-5IAk,11208
+struai/_client.py,sha256=jTRb8UbAwgFyfGoSHfO5_3QMgiHbew-s7n3bbabIZ18,3541
+struai/_exceptions.py,sha256=GK0aVnOdkmcFaO16e0lzj9S2NmnTjQiy5H6nH6yDV24,2021
+struai/_version.py,sha256=b49QzOi1rklaJqTPnxwmLbtWwDKXokP9JbtqeJykzp8,49
+struai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+struai/models/__init__.py,sha256=qA_e6ZGG7Uhq9aLX2LeXlnUuV8VE6ur3OCFOPcj1oco,1220
+struai/models/common.py,sha256=qRehKPxDWPG34_LHhxu7T93Wqz1uRRQzIErQhraYAt4,598
+struai/models/drawings.py,sha256=oTvN0ZID1TdUsR1BkJ-6M87nFbQwsYyePd0xoxKns9I,1634
+struai/models/entities.py,sha256=4MHfhCn4g0n6d-6asASklbcmbJcIPg91rMBc5f56Zm8,1348
+struai/models/projects.py,sha256=_VYinZrwWCmv3SQy7iZFPBtHcGsZe3XxI1i9SCk0EfY,1654
+struai/models/search.py,sha256=UL3RYsAidS1zdV4lKiwNcHcAIjzyi8dgAQodowjzh58,1324
+struai/resources/__init__.py,sha256=_JT_8OrvVcJ4yHmvZnsqYj4ej8pVg5lKSPKIcgQTCZk,183
+struai/resources/drawings.py,sha256=b3C7CHS-p8CpGBtKl_dvLeUFDJ3g8HjlLBA4II0HsT4,3957
+struai/resources/projects.py,sha256=_KRHO6ByBN_W-STlaAhVljLt9F3Yh9xzV5Piahc50FQ,19501
+struai-0.2.0.dist-info/METADATA,sha256=aILI4YDk7L0ZjyjOuA_8mmG0gAXSDV6DO5WEKz0S2iU,4460
+struai-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+struai-0.2.0.dist-info/RECORD,,

struai-0.1.0.dist-info/RECORD

@@ -1,18 +0,0 @@
-struai/__init__.py,sha256=FsxjwfQOKDXzcnX8TqfWgTOqEbe-rW7zf8hwsqSMq0c,2190
-struai/_base.py,sha256=IPhrlgoqf7RZOh9xy2O6o9qMazEj0-RkI-dXPGzk76U,10795
-struai/_client.py,sha256=AmxmJ0ZSPlYlfqf4WHRaR_PQ1u5m3DCE59pLrf8j-7s,3483
-struai/_exceptions.py,sha256=GK0aVnOdkmcFaO16e0lzj9S2NmnTjQiy5H6nH6yDV24,2021
-struai/_version.py,sha256=PrqPeNLB_vK7o-1a2bBnGr1qOX4JyRedkdthES3yNbw,49
-struai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-struai/models/__init__.py,sha256=ddFoRRUY32xeHnjI0twEIRtXNh1huyWwggE1sY-y2e4,1144
-struai/models/common.py,sha256=qRehKPxDWPG34_LHhxu7T93Wqz1uRRQzIErQhraYAt4,598
-struai/models/drawings.py,sha256=oTvN0ZID1TdUsR1BkJ-6M87nFbQwsYyePd0xoxKns9I,1634
-struai/models/entities.py,sha256=dlQJqnY1GbVZOy74Rtt7J2JGxtUI7pWyifAO3whmNP0,753
-struai/models/projects.py,sha256=_VYinZrwWCmv3SQy7iZFPBtHcGsZe3XxI1i9SCk0EfY,1654
-struai/models/search.py,sha256=UL3RYsAidS1zdV4lKiwNcHcAIjzyi8dgAQodowjzh58,1324
-struai/resources/__init__.py,sha256=_JT_8OrvVcJ4yHmvZnsqYj4ej8pVg5lKSPKIcgQTCZk,183
-struai/resources/drawings.py,sha256=b3C7CHS-p8CpGBtKl_dvLeUFDJ3g8HjlLBA4II0HsT4,3957
-struai/resources/projects.py,sha256=3HMLbZmhcPQq51uTT3Y21Rx48PDKXF_-ESPDJYBd5tw,19453
-struai-0.1.0.dist-info/METADATA,sha256=hVEKAZoNtYqtqz_c95ouj7GOmY7jKPVvrfn2tYe3iS0,4309
-struai-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-struai-0.1.0.dist-info/RECORD,,