zep-python/src/zep_cloud/thread/message/client.py at main · getzep/zep-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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
# This file was auto-generated by Fern from our API Definition.

import typing

from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
from ...core.request_options import RequestOptions
from ...types.message import Message
from .raw_client import AsyncRawMessageClient, RawMessageClient

# this is used as the default value for optional parameters
OMIT = typing.cast(typing.Any, ...)


class MessageClient:
    def __init__(self, *, client_wrapper: SyncClientWrapper):
        self._raw_client = RawMessageClient(client_wrapper=client_wrapper)

    @property
    def with_raw_response(self) -> RawMessageClient:
        """
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        RawMessageClient
        """
        return self._raw_client

    def update(
        self,
        message_uuid: str,
        *,
        metadata: typing.Dict[str, typing.Optional[typing.Any]],
        request_options: typing.Optional[RequestOptions] = None,
    ) -> Message:
        """
        Updates a message.

        Parameters
        ----------
        message_uuid : str
            The UUID of the message.

        metadata : typing.Dict[str, typing.Optional[typing.Any]]

        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        Message
            The updated message.

        Examples
        --------
        from zep_cloud import Zep

        client = Zep(
            api_key="YOUR_API_KEY",
        )
        client.thread.message.update(
            message_uuid="messageUUID",
            metadata={"key": "value"},
        )
        """
        _response = self._raw_client.update(message_uuid, metadata=metadata, request_options=request_options)
        return _response.data


class AsyncMessageClient:
    def __init__(self, *, client_wrapper: AsyncClientWrapper):
        self._raw_client = AsyncRawMessageClient(client_wrapper=client_wrapper)

    @property
    def with_raw_response(self) -> AsyncRawMessageClient:
        """
        Retrieves a raw implementation of this client that returns raw responses.

        Returns
        -------
        AsyncRawMessageClient
        """
        return self._raw_client

    async def update(
        self,
        message_uuid: str,
        *,
        metadata: typing.Dict[str, typing.Optional[typing.Any]],
        request_options: typing.Optional[RequestOptions] = None,
    ) -> Message:
        """
        Updates a message.

        Parameters
        ----------
        message_uuid : str
            The UUID of the message.

        metadata : typing.Dict[str, typing.Optional[typing.Any]]

        request_options : typing.Optional[RequestOptions]
            Request-specific configuration.

        Returns
        -------
        Message
            The updated message.

        Examples
        --------
        import asyncio

        from zep_cloud import AsyncZep

        client = AsyncZep(
            api_key="YOUR_API_KEY",
        )


        async def main() -> None:
            await client.thread.message.update(
                message_uuid="messageUUID",
                metadata={"key": "value"},
            )


        asyncio.run(main())
        """
        _response = await self._raw_client.update(message_uuid, metadata=metadata, request_options=request_options)
        return _response.data