arbi 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

arbi/resources/api/workspace.py

@@ -6,7 +6,7 @@ from typing import Optional
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -61,19 +61,22 @@ class WorkspaceResource(SyncAPIResource):
         self,
         workspace_ext_id: str,
         *,
-        description: Optional[str] | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | Omit = omit,
+        is_public: Optional[bool] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceResponse:
-        """Update workspace metadata such as name or description.
+        """Update workspace metadata such as name, description, or public status.
 
-        Changes are persisted to
-        the database.
+        Changes
+        are persisted to the database.
+
+        Only developers can change the is_public field.
 
         Args:
           extra_headers: Send extra headers
@@ -91,6 +94,7 @@ class WorkspaceResource(SyncAPIResource):
             body=maybe_transform(
                 {
                     "description": description,
+                    "is_public": is_public,
                     "name": name,
                 },
                 workspace_update_params.WorkspaceUpdateParams,
@@ -110,7 +114,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceDeleteResponse:
         """Delete a workspace.
 
@@ -142,19 +146,29 @@ class WorkspaceResource(SyncAPIResource):
         self,
         *,
         name: str,
-        description: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | Omit = omit,
+        is_public: bool | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceResponse:
         """Create a new workspace with encryption and access controls.
 
         Sets up vector
         storage and associates the creator as the initial workspace user.
 
+        Public workspaces are visible to all users and grant non-members limited access:
+
+        - Non-members can view shared documents and tags
+        - Non-members can create conversations and send messages
+        - Only members can upload documents
+        - Only members can see the member list
+
+        Only users with developer flag can create public workspaces.
+
         Args:
           extra_headers: Send extra headers
 
@@ -170,6 +184,7 @@ class WorkspaceResource(SyncAPIResource):
                 {
                     "name": name,
                     "description": description,
+                    "is_public": is_public,
                 },
                 workspace_create_protected_params.WorkspaceCreateProtectedParams,
             ),
@@ -188,7 +203,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetConversationsResponse:
         """
         Retrieve conversations for a workspace where the current user is:
@@ -232,7 +247,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetDoctagsResponse:
         """Get all doctags (document-tag associations) in a given workspace.
 
@@ -267,7 +282,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetDocumentsResponse:
         """Retrieve all documents in a workspace with proper access controls.
 
@@ -302,7 +317,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetStatsResponse:
         """
         Retrieves conversation and document counts for a specific workspace.
@@ -335,7 +350,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetTagsResponse:
         """
         Get all tags in a given workspace created by the current user.
@@ -368,12 +383,12 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetUsersResponse:
         """Retrieve users with access to a specific workspace.
 
-        Leverages RLS to enforce
-        appropriate access controls.
+        RLS handles access control:
+        members can view private workspaces, anyone can view public workspaces.
 
         Args:
           extra_headers: Send extra headers
@@ -404,7 +419,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceRemoveUserResponse:
         """
         Remove a user from a workspace.
@@ -441,7 +456,7 @@ class WorkspaceResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceShareResponse:
         """Share a workspace with another user via their email address.
 
@@ -493,19 +508,22 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         self,
         workspace_ext_id: str,
         *,
-        description: Optional[str] | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | Omit = omit,
+        is_public: Optional[bool] | Omit = omit,
+        name: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceResponse:
-        """Update workspace metadata such as name or description.
+        """Update workspace metadata such as name, description, or public status.
 
-        Changes are persisted to
-        the database.
+        Changes
+        are persisted to the database.
+
+        Only developers can change the is_public field.
 
         Args:
           extra_headers: Send extra headers
@@ -523,6 +541,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
             body=await async_maybe_transform(
                 {
                     "description": description,
+                    "is_public": is_public,
                     "name": name,
                 },
                 workspace_update_params.WorkspaceUpdateParams,
@@ -542,7 +561,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceDeleteResponse:
         """Delete a workspace.
 
@@ -574,19 +593,29 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         self,
         *,
         name: str,
-        description: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | Omit = omit,
+        is_public: bool | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceResponse:
         """Create a new workspace with encryption and access controls.
 
         Sets up vector
         storage and associates the creator as the initial workspace user.
 
+        Public workspaces are visible to all users and grant non-members limited access:
+
+        - Non-members can view shared documents and tags
+        - Non-members can create conversations and send messages
+        - Only members can upload documents
+        - Only members can see the member list
+
+        Only users with developer flag can create public workspaces.
+
         Args:
           extra_headers: Send extra headers
 
@@ -602,6 +631,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
                 {
                     "name": name,
                     "description": description,
+                    "is_public": is_public,
                 },
                 workspace_create_protected_params.WorkspaceCreateProtectedParams,
             ),
@@ -620,7 +650,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetConversationsResponse:
         """
         Retrieve conversations for a workspace where the current user is:
@@ -664,7 +694,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetDoctagsResponse:
         """Get all doctags (document-tag associations) in a given workspace.
 
@@ -699,7 +729,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetDocumentsResponse:
         """Retrieve all documents in a workspace with proper access controls.
 
@@ -734,7 +764,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetStatsResponse:
         """
         Retrieves conversation and document counts for a specific workspace.
@@ -767,7 +797,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetTagsResponse:
         """
         Get all tags in a given workspace created by the current user.
@@ -800,12 +830,12 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceGetUsersResponse:
         """Retrieve users with access to a specific workspace.
 
-        Leverages RLS to enforce
-        appropriate access controls.
+        RLS handles access control:
+        members can view private workspaces, anyone can view public workspaces.
 
         Args:
           extra_headers: Send extra headers
@@ -836,7 +866,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceRemoveUserResponse:
         """
         Remove a user from a workspace.
@@ -875,7 +905,7 @@ class AsyncWorkspaceResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> WorkspaceShareResponse:
         """Share a workspace with another user via their email address.
 

arbi/types/api/__init__.py

@@ -45,6 +45,7 @@ from .config_retrieve_response import ConfigRetrieveResponse as ConfigRetrieveRe
 from .document_delete_response import DocumentDeleteResponse as DocumentDeleteResponse
 from .document_update_response import DocumentUpdateResponse as DocumentUpdateResponse
 from .tag_apply_to_docs_params import TagApplyToDocsParams as TagApplyToDocsParams
+from .user_verify_email_params import UserVerifyEmailParams as UserVerifyEmailParams
 from .workspace_share_response import WorkspaceShareResponse as WorkspaceShareResponse
 from .assistant_retrieve_params import AssistantRetrieveParams as AssistantRetrieveParams
 from .health_check_app_response import HealthCheckAppResponse as HealthCheckAppResponse
@@ -52,6 +53,7 @@ from .workspace_delete_response import WorkspaceDeleteResponse as WorkspaceDelet
 from .document_get_tags_response import DocumentGetTagsResponse as DocumentGetTagsResponse
 from .health_get_models_response import HealthGetModelsResponse as HealthGetModelsResponse
 from .tag_apply_to_docs_response import TagApplyToDocsResponse as TagApplyToDocsResponse
+from .user_verify_email_response import UserVerifyEmailResponse as UserVerifyEmailResponse
 from .conversation_share_response import ConversationShareResponse as ConversationShareResponse
 from .model_citation_config_param import ModelCitationConfigParam as ModelCitationConfigParam
 from .tag_remove_from_docs_params import TagRemoveFromDocsParams as TagRemoveFromDocsParams
@@ -69,7 +71,10 @@ from .user_list_workspaces_response import UserListWorkspacesResponse as UserLis
 from .health_check_services_response import HealthCheckServicesResponse as HealthCheckServicesResponse
 from .workspace_get_doctags_response import WorkspaceGetDoctagsResponse as WorkspaceGetDoctagsResponse
 from .workspace_remove_user_response import WorkspaceRemoveUserResponse as WorkspaceRemoveUserResponse
+from .document_upload_from_url_params import DocumentUploadFromURLParams as DocumentUploadFromURLParams
+from .health_retrieve_status_response import HealthRetrieveStatusResponse as HealthRetrieveStatusResponse
 from .conversation_update_title_params import ConversationUpdateTitleParams as ConversationUpdateTitleParams
+from .health_retrieve_version_response import HealthRetrieveVersionResponse as HealthRetrieveVersionResponse
 from .workspace_get_documents_response import WorkspaceGetDocumentsResponse as WorkspaceGetDocumentsResponse
 from .workspace_create_protected_params import WorkspaceCreateProtectedParams as WorkspaceCreateProtectedParams
 from .conversation_update_title_response import ConversationUpdateTitleResponse as ConversationUpdateTitleResponse

arbi/types/api/assistant_query_params.py

@@ -2,15 +2,17 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Iterable, Optional
+from typing import Dict, Union, Iterable, Optional
 from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
+from ..._types import SequenceNotStr
 from ..chunk_param import ChunkParam
 
 __all__ = [
     "AssistantQueryParams",
     "Tools",
     "ToolsModelCitationTool",
+    "ToolsModelCitationToolToolResponses",
     "ToolsRetrievalChunkToolInput",
     "ToolsRetrievalFullContextToolInput",
 ]
@@ -28,12 +30,22 @@ class AssistantQueryParams(TypedDict, total=False):
     tools: Dict[str, Tools]
 
 
+class ToolsModelCitationToolToolResponses(TypedDict, total=False):
+    chunk_ids: Required[SequenceNotStr[str]]
+
+    offset_end: Required[int]
+
+    offset_start: Required[int]
+
+    statement: Required[str]
+
+
 class ToolsModelCitationTool(TypedDict, total=False):
     description: str
 
     name: Literal["model_citation"]
 
-    tool_responses: Dict[str, List[str]]
+    tool_responses: Dict[str, ToolsModelCitationToolToolResponses]
 
 
 class ToolsRetrievalChunkToolInput(TypedDict, total=False):
@@ -41,7 +53,7 @@ class ToolsRetrievalChunkToolInput(TypedDict, total=False):
 
     name: Literal["retrieval_chunk"]
 
-    tool_args: Dict[str, List[str]]
+    tool_args: Dict[str, SequenceNotStr[str]]
 
     tool_responses: Dict[str, Iterable[ChunkParam]]
 
@@ -51,7 +63,7 @@ class ToolsRetrievalFullContextToolInput(TypedDict, total=False):
 
     name: Literal["retrieval_full_context"]
 
-    tool_args: Dict[str, List[str]]
+    tool_args: Dict[str, object]
 
     tool_responses: Dict[str, Iterable[ChunkParam]]
 

arbi/types/api/assistant_retrieve_params.py

@@ -2,15 +2,17 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Iterable, Optional
+from typing import Dict, Union, Iterable, Optional
 from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
+from ..._types import SequenceNotStr
 from ..chunk_param import ChunkParam
 
 __all__ = [
     "AssistantRetrieveParams",
     "Tools",
     "ToolsModelCitationTool",
+    "ToolsModelCitationToolToolResponses",
     "ToolsRetrievalChunkToolInput",
     "ToolsRetrievalFullContextToolInput",
 ]
@@ -28,12 +30,22 @@ class AssistantRetrieveParams(TypedDict, total=False):
     tools: Dict[str, Tools]
 
 
+class ToolsModelCitationToolToolResponses(TypedDict, total=False):
+    chunk_ids: Required[SequenceNotStr[str]]
+
+    offset_end: Required[int]
+
+    offset_start: Required[int]
+
+    statement: Required[str]
+
+
 class ToolsModelCitationTool(TypedDict, total=False):
     description: str
 
     name: Literal["model_citation"]
 
-    tool_responses: Dict[str, List[str]]
+    tool_responses: Dict[str, ToolsModelCitationToolToolResponses]
 
 
 class ToolsRetrievalChunkToolInput(TypedDict, total=False):
@@ -41,7 +53,7 @@ class ToolsRetrievalChunkToolInput(TypedDict, total=False):
 
     name: Literal["retrieval_chunk"]
 
-    tool_args: Dict[str, List[str]]
+    tool_args: Dict[str, SequenceNotStr[str]]
 
     tool_responses: Dict[str, Iterable[ChunkParam]]
 
@@ -51,7 +63,7 @@ class ToolsRetrievalFullContextToolInput(TypedDict, total=False):
 
     name: Literal["retrieval_full_context"]
 
-    tool_args: Dict[str, List[str]]
+    tool_args: Dict[str, object]
 
     tool_responses: Dict[str, Iterable[ChunkParam]]
 

arbi/types/api/config_create_params.py

@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from typing import Optional
-from typing_extensions import Annotated, TypedDict
+from typing_extensions import Literal, Annotated, TypedDict
 
 from ..._utils import PropertyInfo
 from .parser_config_param import ParserConfigParam
@@ -16,10 +16,14 @@ from .title_llm_config_param import TitleLlmConfigParam
 from .model_citation_config_param import ModelCitationConfigParam
 from .document_date_extractor_llm_config_param import DocumentDateExtractorLlmConfigParam
 
-__all__ = ["ConfigCreateParams"]
+__all__ = ["ConfigCreateParams", "AgentLlm", "Agents"]
 
 
 class ConfigCreateParams(TypedDict, total=False):
+    agent_llm: Annotated[Optional[AgentLlm], PropertyInfo(alias="AgentLLM")]
+
+    agents: Annotated[Optional[Agents], PropertyInfo(alias="Agents")]
+
     chunker: Annotated[Optional[ChunkerConfigParam], PropertyInfo(alias="Chunker")]
 
     document_date_extractor_llm: Annotated[
@@ -43,3 +47,64 @@ class ConfigCreateParams(TypedDict, total=False):
     title: str
 
     title_llm: Annotated[Optional[TitleLlmConfigParam], PropertyInfo(alias="TitleLLM")]
+
+
+class AgentLlm(TypedDict, total=False):
+    api_type: Annotated[Literal["local", "remote"], PropertyInfo(alias="API_TYPE")]
+    """The inference type (local or remote)."""
+
+    enabled: Annotated[bool, PropertyInfo(alias="ENABLED")]
+    """Whether to use agent mode for queries."""
+
+    max_char_size_to_answer: Annotated[int, PropertyInfo(alias="MAX_CHAR_SIZE_TO_ANSWER")]
+    """Maximum character size for history."""
+
+    max_context_tokens: Annotated[int, PropertyInfo(alias="MAX_CONTEXT_TOKENS")]
+    """
+    Maximum tokens for gathered context (applies to evidence buffer and final
+    query).
+    """
+
+    max_iterations: Annotated[int, PropertyInfo(alias="MAX_ITERATIONS")]
+    """Maximum agent loop iterations."""
+
+    max_tokens: Annotated[int, PropertyInfo(alias="MAX_TOKENS")]
+    """Maximum tokens for planning decisions."""
+
+    model_name: Annotated[str, PropertyInfo(alias="MODEL_NAME")]
+    """The name of the model to be used."""
+
+    show_interim_steps: Annotated[bool, PropertyInfo(alias="SHOW_INTERIM_STEPS")]
+    """Whether to show agent's intermediate steps."""
+
+    system_instruction: Annotated[str, PropertyInfo(alias="SYSTEM_INSTRUCTION")]
+    """The system instruction for agent planning."""
+
+    temperature: Annotated[float, PropertyInfo(alias="TEMPERATURE")]
+    """Temperature for agent decisions."""
+
+
+class Agents(TypedDict, total=False):
+    agent_model_name: Annotated[str, PropertyInfo(alias="AGENT_MODEL_NAME")]
+    """The name of the model to be used for the agent."""
+
+    agent_prompt: Annotated[str, PropertyInfo(alias="AGENT_PROMPT")]
+
+    enabled: Annotated[bool, PropertyInfo(alias="ENABLED")]
+    """Whether to use agents mode for queries."""
+
+    llm_page_filter_model_name: Annotated[str, PropertyInfo(alias="LLM_PAGE_FILTER_MODEL_NAME")]
+    """The name of the model to be used for the llm page filter model."""
+
+    llm_page_filter_prompt: Annotated[str, PropertyInfo(alias="LLM_PAGE_FILTER_PROMPT")]
+
+    llm_page_filter_temperature: Annotated[float, PropertyInfo(alias="LLM_PAGE_FILTER_TEMPERATURE")]
+    """Temperature value for randomness."""
+
+    llm_summarise_model_name: Annotated[str, PropertyInfo(alias="LLM_SUMMARISE_MODEL_NAME")]
+    """The name of the model to be used for the llm summarise model."""
+
+    llm_summarise_prompt: Annotated[str, PropertyInfo(alias="LLM_SUMMARISE_PROMPT")]
+
+    llm_summarise_temperature: Annotated[float, PropertyInfo(alias="LLM_SUMMARISE_TEMPERATURE")]
+    """Temperature value for randomness."""