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

struai/__init__.py

@@ -17,50 +17,52 @@ Example:
     >>> results = project.search("W12x26 beam connections")
 """
 
-from ._version import __version__
-from ._client import StruAI, AsyncStruAI
+from ._client import AsyncStruAI, StruAI
 from ._exceptions import (
-    StruAIError,
     APIError,
     AuthenticationError,
-    PermissionDeniedError,
+    ConnectionError,
+    InternalServerError,
+    JobFailedError,
     NotFoundError,
-    ValidationError,
+    PermissionDeniedError,
     RateLimitError,
-    InternalServerError,
+    StruAIError,
     TimeoutError,
-    ConnectionError,
-    JobFailedError,
+    ValidationError,
 )
+from ._version import __version__
 
 # Re-export commonly used models
 from .models import (
-    # Common
-    Point,
+    Annotations,
     BBox,
-    TextSpan,
+    DetailTag,
     Dimensions,
     # Tier 1 - Drawings
     DrawingResult,
-    Annotations,
+    # Entities
+    Entity,
+    EntityListItem,
+    EntityRelation,
+    Fact,
+    JobStatus,
     Leader,
-    SectionTag,
-    DetailTag,
-    RevisionTriangle,
-    RevisionCloud,
-    TitleBlock,
+    # Common
+    Point,
     # Tier 2 - Projects
     Project,
-    Sheet,
-    JobStatus,
-    SheetResult,
+    QueryResponse,
+    RevisionCloud,
+    RevisionTriangle,
+    SearchHit,
     # Search
     SearchResponse,
-    SearchHit,
-    QueryResponse,
-    # Entities
-    Entity,
-    Fact,
+    SectionTag,
+    Sheet,
+    SheetResult,
+    TextSpan,
+    TitleBlock,
 )
 
 __all__ = [
@@ -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

@@ -1,9 +1,10 @@
 """Main StruAI client classes."""
+
 import os
 from functools import cached_property
 from typing import Optional
 
-from ._base import AsyncBaseClient, BaseClient, DEFAULT_BASE_URL, DEFAULT_TIMEOUT
+from ._base import DEFAULT_BASE_URL, DEFAULT_TIMEOUT, AsyncBaseClient, BaseClient
 from ._exceptions import StruAIError
 from .resources.drawings import AsyncDrawings, Drawings
 from .resources.projects import AsyncProjects, Projects
@@ -14,7 +15,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__ = "1.0.1"

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/common.py

@@ -1,5 +1,5 @@
 """Common types shared across models."""
-from typing import List, Tuple
+from typing import Tuple
 
 from pydantic import BaseModel
 

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-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: struai
-Version: 0.1.0
+Version: 1.0.1
 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-1.0.1.dist-info/RECORD

@@ -0,0 +1,18 @@
+struai/__init__.py,sha256=3YTAjICfleWQh33O7y-yVqAJ8k9Pn2eU2epzrroKqt4,2274
+struai/_base.py,sha256=JiAxWI1PMmIFVr9v_hBABra8m2fRkEvBp7d2Vc-5IAk,11208
+struai/_client.py,sha256=WOcGC7LjaG469shx1YX2GCt0yJs-Ew-vOrxsqUmVmHM,3542
+struai/_exceptions.py,sha256=GK0aVnOdkmcFaO16e0lzj9S2NmnTjQiy5H6nH6yDV24,2021
+struai/_version.py,sha256=da8k-y4StnRTawPjG6KLx2cylvLgrpCSR6eZl8UgnKA,49
+struai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+struai/models/__init__.py,sha256=qA_e6ZGG7Uhq9aLX2LeXlnUuV8VE6ur3OCFOPcj1oco,1220
+struai/models/common.py,sha256=0Ox9PiuyRoyWH5vbSTfjm2nCWt_pblWH7kJso8nMVuE,592
+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-1.0.1.dist-info/METADATA,sha256=pVsCT9ov986VJ3flFsJhCGwMIGrYlO_ITbdoPJ1_gV8,4460
+struai-1.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+struai-1.0.1.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,,