2424from .clients .budget import BudgetClient
2525from .clients .tool import ToolClient
2626from .clients .analytics import AnalyticsClient
27+ from .clients .context import ContextClient
28+ from .clients .documents import DocumentClient
2729from .exceptions import GovernsAIError
2830
2931
@@ -32,7 +34,7 @@ class GovernsAIConfig:
3234 """Configuration for GovernsAI client."""
3335
3436 api_key : str
35- base_url : str = "http ://localhost:3002 "
37+ base_url : str = "https ://api.governsai.com "
3638 org_id : str = ""
3739 timeout : int = 30000
3840 retries : int = 3
@@ -83,7 +85,7 @@ def __init__(
8385 else :
8486 self .config = GovernsAIConfig (
8587 api_key = api_key or os .getenv ("GOVERNS_API_KEY" , "" ),
86- base_url = base_url or os .getenv ("GOVERNS_BASE_URL" , "http ://localhost:3002 " ),
88+ base_url = base_url or os .getenv ("GOVERNS_BASE_URL" , "https ://api.governsai.com " ),
8789 org_id = org_id or os .getenv ("GOVERNS_ORG_ID" , "" ),
8890 timeout = timeout or int (os .getenv ("GOVERNS_TIMEOUT" , "30000" )),
8991 retries = retries or int (os .getenv ("GOVERNS_RETRIES" , "3" )),
@@ -116,6 +118,8 @@ def __init__(
116118 self .budget = BudgetClient (self .http_client , self .logger )
117119 self .tools = ToolClient (self .http_client , self .logger )
118120 self .analytics = AnalyticsClient (self .http_client , self .logger )
121+ self .context = ContextClient (self .http_client , self .logger )
122+ self .documents = DocumentClient (self .http_client , self .logger )
119123
120124 @classmethod
121125 def from_env (cls ) -> "GovernsAIClient" :
0 commit comments