Aller au contenu

Conversations Module

The conversations module provides functionality for managing conversation in Forra.

ConversationsAPI

Source code in forrasdk/api/conversations.py 
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
class ConversationsAPI:
    def __init__(self, base_url: str, headers: dict, retry_strategy: Callable) -> None:
        self._base_url = base_url
        self._headers = headers
        self._retry_strategy = retry_strategy

    def create(
        self,
        messages: Optional[list[ConversationMessage]] = None,
        assistant_id: Optional[str] = None,
        model: Optional[str] = None,
        title: Optional[str] = None,
        time_zone_offset: str = "-0400",
        user_data: Optional[dict] = None,
    ) -> ConversationResponse:
        """
        Create a new conversation.

        Args:
            messages (Optional[list[ConversationMessage]]): The list of messages to start the conversation with.
            time_zone_offset (str, optional): Time zone offset for the conversation. Defaults to "-0400".
            assistant_id (Optional[str]): The ID of the assistant to associate with the conversation. If None, creates a generic conversation.
            title (Optional[str]): Title of the conversation. If None, no title is set.
            model (Optional[str]): The model to use for the conversation. If None, uses the default model.
            user_data (Optional[dict]): Additional user data for the conversation.

        Returns:
            ConversationResponse: The response from the Scout API after creating the conversation.

        Raises:
            ValueError: If the response is invalid or cannot be validated.
        """

        request_payload = CreateConversationRequest(
            payload=messages or [],
            time_zone_offset=time_zone_offset,
            assistant_id=assistant_id,
            title=title,
            model=model,
            user_data=user_data,
        )

        response, status_code = RequestUtils.post(
            url=f"{self._base_url}/api/conversations/",
            headers=self._headers,
            json_payload=request_payload.model_dump(exclude_none=True),
            retry_strategy=self._retry_strategy,
        )
        return ConversationResponse.model_validate(response)

    def delete(self, conversation_id: str) -> DeleteConversationResponse:
        """
        Delete a conversation.

        Args:
            conversation_id (str): The ID of the conversation to delete.

        Returns:
            DeleteConversationResponse: The response containing the ID of the deleted conversation.

        Raises:
            ValueError: If the response is invalid or cannot be validated.
        """
        response, status_code = RequestUtils.delete(
            url=f"{self._base_url}/api/conversations/{conversation_id}",
            headers=self._headers,
            retry_strategy=self._retry_strategy,
        )
        return DeleteConversationResponse(id=response)

    def list_all(
        self,
        limit: int = 200,
    ) -> List[ConversationResponse]:
        """
        Retrieve all conversations accessible to the current user.

        Args:
            limit (int): Maximum number of conversations to return. Defaults to 200.

        Returns:
            List[ConversationResponse]: A list of conversations accessible to the current user.
        """
        response, _ = RequestUtils.get(
            url=f"{self._base_url}/api/conversations",
            headers=self._headers,
            retry_strategy=self._retry_strategy,
            params={"limit": limit},
        )
        return [ConversationResponse.model_validate(c) for c in response]

    def get_signed_upload_url(
        self,
        conversation_id: str,
        file_path: str,
    ) -> SignedUploadUrlResponse:
        """
        Generate a signed upload URL for uploading a file to a specific conversation.

        Args:
            conversation_id (str): The ID of the conversation to associate the file with.
            file_path (str): The local file path of the file to be uploaded.

        Returns:
            SignedUploadUrlResponse: The response object containing the signed upload URL.

        Raises:
            ValueError: If the response is invalid or cannot be validated.
        """
        payload = {"file_path": file_path}

        response, status_code = RequestUtils.post(
            url=f"{self._base_url}/api/conversations/{conversation_id}/signed-upload-url",
            headers=self._headers,
            json_payload=payload,
            retry_strategy=self._retry_strategy,
        )
        try:
            return SignedUploadUrlResponse.model_validate(response)
        except Exception as e:
            raise ValueError(f"Invalid response: {response}") from e

    def get_signed_url(
        self,
        conversation_id: str,
        file_path: str,
    ) -> SignedUrlResponse:
        """
        Generate a signed URL for accessing a file associated with a specific conversation.

        Args:
            conversation_id (str): The ID of the conversation.
            file_path (str): The relative path of the file within the conversation.

        Returns:
            SignedUrlResponse: The response object containing the signed URL for the file.

        Raises:
            ValueError: If the response is invalid or cannot be validated.
        """
        response, status_code = RequestUtils.get(
            url=f"{self._base_url}/api/protected/conversations/{conversation_id}/{file_path}",
            headers=self._headers,
            retry_strategy=self._retry_strategy,
        )

        return SignedUrlResponse.model_validate(response)

    def get_user_data(
        self,
        conversation_id: str,
        model_class: Optional[Type[T]] = None,
    ) -> T | dict:
        """
        Get user data for a specific conversation.

        Args:
            conversation_id (str): The ID of the conversation.
            model_class (Optional[Type[T]]): Pydantic model class to validate the user data.
                If None, returns raw dict.

        Returns:
            T | dict: The user data, optionally validated with the provided model class.

        Raises:
            ValueError: If the response is invalid or cannot be validated.
        """
        response, status_code = RequestUtils.get(
            url=f"{self._base_url}/api/conversations/{conversation_id}",
            headers=self._headers,
            retry_strategy=self._retry_strategy,
        )

        conversation_response = ConversationResponse.model_validate(response)
        user_data = conversation_response.user_data or {}

        if model_class:
            return model_class.model_validate(user_data)
        return user_data

    def update_user_data(
        self,
        conversation_id: str,
        user_data: BaseModel | dict,
    ) -> ConversationResponse:
        """
        Update user data for a specific conversation.

        Args:
            conversation_id (str): The ID of the conversation.
            user_data (BaseModel | dict): The user data to update.

        Returns:
            ConversationResponse: The updated conversation response from the API.

        Raises:
            ValueError: If the response is invalid or cannot be validated.
        """
        request_data = (
            user_data.model_dump() if isinstance(user_data, BaseModel) else user_data
        )

        response, status_code = RequestUtils.post(
            url=f"{self._base_url}/api/conversations/{conversation_id}/user_data",
            headers=self._headers,
            json_payload=request_data,
            retry_strategy=self._retry_strategy,
        )

        return ConversationResponse.model_validate(response)

    def update_user_data_key(
        self,
        conversation_id: str,
        key: str,
        value: Any,
    ) -> ConversationResponse:
        payload = value.model_dump() if isinstance(value, BaseModel) else value
        response, status_code = RequestUtils.post(
            url=f"{self._base_url}/api/conversations/{conversation_id}/user_data/{key}",
            headers=self._headers,
            json_payload={"data": payload},
            retry_strategy=self._retry_strategy,
        )
        return ConversationResponse.model_validate(response)

create(messages=None, assistant_id=None, model=None, title=None, time_zone_offset='-0400', user_data=None)

Create a new conversation.

Parameters:

Name Type Description Default
messages Optional[list[ConversationMessage]]

The list of messages to start the conversation with.

 None
time_zone_offset str

Time zone offset for the conversation. Defaults to "-0400".

 '-0400'
assistant_id Optional[str]

The ID of the assistant to associate with the conversation. If None, creates a generic conversation.

 None
title Optional[str]

Title of the conversation. If None, no title is set.

 None
model Optional[str]

The model to use for the conversation. If None, uses the default model.

 None
user_data Optional[dict]

Additional user data for the conversation.

 None

Returns:

Name Type Description
ConversationResponse ConversationResponse

The response from the Scout API after creating the conversation.

Raises:

Type Description
ValueError

If the response is invalid or cannot be validated.
Source code in forrasdk/api/conversations.py 
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
def create(
    self,
    messages: Optional[list[ConversationMessage]] = None,
    assistant_id: Optional[str] = None,
    model: Optional[str] = None,
    title: Optional[str] = None,
    time_zone_offset: str = "-0400",
    user_data: Optional[dict] = None,
) -> ConversationResponse:
    """
    Create a new conversation.

    Args:
        messages (Optional[list[ConversationMessage]]): The list of messages to start the conversation with.
        time_zone_offset (str, optional): Time zone offset for the conversation. Defaults to "-0400".
        assistant_id (Optional[str]): The ID of the assistant to associate with the conversation. If None, creates a generic conversation.
        title (Optional[str]): Title of the conversation. If None, no title is set.
        model (Optional[str]): The model to use for the conversation. If None, uses the default model.
        user_data (Optional[dict]): Additional user data for the conversation.

    Returns:
        ConversationResponse: The response from the Scout API after creating the conversation.

    Raises:
        ValueError: If the response is invalid or cannot be validated.
    """

    request_payload = CreateConversationRequest(
        payload=messages or [],
        time_zone_offset=time_zone_offset,
        assistant_id=assistant_id,
        title=title,
        model=model,
        user_data=user_data,
    )

    response, status_code = RequestUtils.post(
        url=f"{self._base_url}/api/conversations/",
        headers=self._headers,
        json_payload=request_payload.model_dump(exclude_none=True),
        retry_strategy=self._retry_strategy,
    )
    return ConversationResponse.model_validate(response)

delete(conversation_id)

Delete a conversation.

Parameters:

Name Type Description Default
conversation_id str

The ID of the conversation to delete.

 required

Returns:

Name Type Description
DeleteConversationResponse DeleteConversationResponse

The response containing the ID of the deleted conversation.

Raises:

Type Description
ValueError

If the response is invalid or cannot be validated.
Source code in forrasdk/api/conversations.py 
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
def delete(self, conversation_id: str) -> DeleteConversationResponse:
    """
    Delete a conversation.

    Args:
        conversation_id (str): The ID of the conversation to delete.

    Returns:
        DeleteConversationResponse: The response containing the ID of the deleted conversation.

    Raises:
        ValueError: If the response is invalid or cannot be validated.
    """
    response, status_code = RequestUtils.delete(
        url=f"{self._base_url}/api/conversations/{conversation_id}",
        headers=self._headers,
        retry_strategy=self._retry_strategy,
    )
    return DeleteConversationResponse(id=response)

get_signed_upload_url(conversation_id, file_path)

Generate a signed upload URL for uploading a file to a specific conversation.

Parameters:

Name Type Description Default
conversation_id str

The ID of the conversation to associate the file with.

 required
file_path str

The local file path of the file to be uploaded.

 required

Returns:

Name Type Description
SignedUploadUrlResponse SignedUploadUrlResponse

The response object containing the signed upload URL.

Raises:

Type Description
ValueError

If the response is invalid or cannot be validated.
Source code in forrasdk/api/conversations.py 
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
def get_signed_upload_url(
    self,
    conversation_id: str,
    file_path: str,
) -> SignedUploadUrlResponse:
    """
    Generate a signed upload URL for uploading a file to a specific conversation.

    Args:
        conversation_id (str): The ID of the conversation to associate the file with.
        file_path (str): The local file path of the file to be uploaded.

    Returns:
        SignedUploadUrlResponse: The response object containing the signed upload URL.

    Raises:
        ValueError: If the response is invalid or cannot be validated.
    """
    payload = {"file_path": file_path}

    response, status_code = RequestUtils.post(
        url=f"{self._base_url}/api/conversations/{conversation_id}/signed-upload-url",
        headers=self._headers,
        json_payload=payload,
        retry_strategy=self._retry_strategy,
    )
    try:
        return SignedUploadUrlResponse.model_validate(response)
    except Exception as e:
        raise ValueError(f"Invalid response: {response}") from e

get_signed_url(conversation_id, file_path)

Generate a signed URL for accessing a file associated with a specific conversation.

Parameters:

Name Type Description Default
conversation_id str

The ID of the conversation.

 required
file_path str

The relative path of the file within the conversation.

 required

Returns:

Name Type Description
SignedUrlResponse SignedUrlResponse

The response object containing the signed URL for the file.

Raises:

Type Description
ValueError

If the response is invalid or cannot be validated.
Source code in forrasdk/api/conversations.py 
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
def get_signed_url(
    self,
    conversation_id: str,
    file_path: str,
) -> SignedUrlResponse:
    """
    Generate a signed URL for accessing a file associated with a specific conversation.

    Args:
        conversation_id (str): The ID of the conversation.
        file_path (str): The relative path of the file within the conversation.

    Returns:
        SignedUrlResponse: The response object containing the signed URL for the file.

    Raises:
        ValueError: If the response is invalid or cannot be validated.
    """
    response, status_code = RequestUtils.get(
        url=f"{self._base_url}/api/protected/conversations/{conversation_id}/{file_path}",
        headers=self._headers,
        retry_strategy=self._retry_strategy,
    )

    return SignedUrlResponse.model_validate(response)

get_user_data(conversation_id, model_class=None)

Get user data for a specific conversation.

Parameters:

Name Type Description Default
conversation_id str

The ID of the conversation.

 required
model_class Optional[Type[T]]

Pydantic model class to validate the user data. If None, returns raw dict.

 None

Returns:

Type Description
T | dict

T | dict: The user data, optionally validated with the provided model class.

Raises:

Type Description
ValueError

If the response is invalid or cannot be validated.
Source code in forrasdk/api/conversations.py 
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
def get_user_data(
    self,
    conversation_id: str,
    model_class: Optional[Type[T]] = None,
) -> T | dict:
    """
    Get user data for a specific conversation.

    Args:
        conversation_id (str): The ID of the conversation.
        model_class (Optional[Type[T]]): Pydantic model class to validate the user data.
            If None, returns raw dict.

    Returns:
        T | dict: The user data, optionally validated with the provided model class.

    Raises:
        ValueError: If the response is invalid or cannot be validated.
    """
    response, status_code = RequestUtils.get(
        url=f"{self._base_url}/api/conversations/{conversation_id}",
        headers=self._headers,
        retry_strategy=self._retry_strategy,
    )

    conversation_response = ConversationResponse.model_validate(response)
    user_data = conversation_response.user_data or {}

    if model_class:
        return model_class.model_validate(user_data)
    return user_data

list_all(limit=200)

Retrieve all conversations accessible to the current user.

Parameters:

Name Type Description Default
limit int

Maximum number of conversations to return. Defaults to 200.

 200

Returns:

Type Description
List[ConversationResponse]

List[ConversationResponse]: A list of conversations accessible to the current user.
Source code in forrasdk/api/conversations.py 
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
def list_all(
    self,
    limit: int = 200,
) -> List[ConversationResponse]:
    """
    Retrieve all conversations accessible to the current user.

    Args:
        limit (int): Maximum number of conversations to return. Defaults to 200.

    Returns:
        List[ConversationResponse]: A list of conversations accessible to the current user.
    """
    response, _ = RequestUtils.get(
        url=f"{self._base_url}/api/conversations",
        headers=self._headers,
        retry_strategy=self._retry_strategy,
        params={"limit": limit},
    )
    return [ConversationResponse.model_validate(c) for c in response]

update_user_data(conversation_id, user_data)

Update user data for a specific conversation.

Parameters:

Name Type Description Default
conversation_id str

The ID of the conversation.

 required
user_data BaseModel | dict

The user data to update.

 required

Returns:

Name Type Description
ConversationResponse ConversationResponse

The updated conversation response from the API.

Raises:

Type Description
ValueError

If the response is invalid or cannot be validated.
Source code in forrasdk/api/conversations.py 
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
def update_user_data(
    self,
    conversation_id: str,
    user_data: BaseModel | dict,
) -> ConversationResponse:
    """
    Update user data for a specific conversation.

    Args:
        conversation_id (str): The ID of the conversation.
        user_data (BaseModel | dict): The user data to update.

    Returns:
        ConversationResponse: The updated conversation response from the API.

    Raises:
        ValueError: If the response is invalid or cannot be validated.
    """
    request_data = (
        user_data.model_dump() if isinstance(user_data, BaseModel) else user_data
    )

    response, status_code = RequestUtils.post(
        url=f"{self._base_url}/api/conversations/{conversation_id}/user_data",
        headers=self._headers,
        json_payload=request_data,
        retry_strategy=self._retry_strategy,
    )

    return ConversationResponse.model_validate(response)