gllm-memory-binary 0.2.0.post1__cp313-cp313-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (50) hide show
  1. gllm_memory/__init__.pyi +8 -0
  2. gllm_memory/backends/__init__.pyi +5 -0
  3. gllm_memory/backends/base.pyi +34 -0
  4. gllm_memory/backends/mem0/__init__.pyi +3 -0
  5. gllm_memory/backends/mem0/adapter.pyi +35 -0
  6. gllm_memory/backends/mem0/config_compiler.pyi +38 -0
  7. gllm_memory/backends/mem0/defaults.pyi +21 -0
  8. gllm_memory/backends/resolver.pyi +29 -0
  9. gllm_memory/builder/__init__.pyi +4 -0
  10. gllm_memory/builder/build_memory_client.pyi +45 -0
  11. gllm_memory/builder/memory_manager_builder.pyi +275 -0
  12. gllm_memory/builder/memory_manager_defaults.pyi +99 -0
  13. gllm_memory/chat_history_manager/__init__.pyi +1 -0
  14. gllm_memory/clients/__init__.pyi +3 -0
  15. gllm_memory/clients/constants.pyi +7 -0
  16. gllm_memory/clients/mem0/__init__.pyi +5 -0
  17. gllm_memory/clients/mem0/core/__init__.pyi +3 -0
  18. gllm_memory/clients/mem0/core/mem0_add_important_policy.pyi +78 -0
  19. gllm_memory/clients/mem0/core/mem0_base_client.pyi +216 -0
  20. gllm_memory/clients/mem0/core/mem0_chunk_creator.pyi +17 -0
  21. gllm_memory/clients/mem0/core/mem0_filter_builder.pyi +47 -0
  22. gllm_memory/clients/mem0/core/mem0_filter_enhancer.pyi +61 -0
  23. gllm_memory/clients/mem0/core/mem0_filter_strategy.pyi +34 -0
  24. gllm_memory/clients/mem0/core/mem0_message_mapper.pyi +21 -0
  25. gllm_memory/clients/mem0/core/mem0_response_processor.pyi +45 -0
  26. gllm_memory/clients/mem0/core/mem0_schema.pyi +34 -0
  27. gllm_memory/clients/mem0/http/__init__.pyi +4 -0
  28. gllm_memory/clients/mem0/http/mem0_client.pyi +32 -0
  29. gllm_memory/clients/mem0/http/mem0_self_hosted_client.pyi +339 -0
  30. gllm_memory/clients/mem0/sdk/__init__.pyi +3 -0
  31. gllm_memory/clients/mem0/sdk/mem0_sdk_client.pyi +88 -0
  32. gllm_memory/clients/mem0/sdk/mem0_sdk_elasticsearch.pyi +182 -0
  33. gllm_memory/clients/mem0/sdk/mem0_sdk_elasticsearch_config.pyi +12 -0
  34. gllm_memory/clients/mem0/sdk/mem0_sdk_gllm_inference.pyi +128 -0
  35. gllm_memory/clients/mem0/sdk/mem0_sdk_gllm_inference_embedding.pyi +67 -0
  36. gllm_memory/clients/mem0/sdk/mem0_sdk_gllm_inference_embedding_normalizer.pyi +24 -0
  37. gllm_memory/clients/mem0/sdk/mem0_sdk_gllm_inference_runtime.pyi +31 -0
  38. gllm_memory/clients/mem0/sdk/mem0_sdk_patches.pyi +175 -0
  39. gllm_memory/clients/memory_client.pyi +158 -0
  40. gllm_memory/clients/utils.pyi +68 -0
  41. gllm_memory/enums.pyi +81 -0
  42. gllm_memory/memory_manager.pyi +175 -0
  43. gllm_memory/schema.pyi +43 -0
  44. gllm_memory/utils/__init__.pyi +0 -0
  45. gllm_memory.cp313-win_amd64.pyd +0 -0
  46. gllm_memory.pyi +106 -0
  47. gllm_memory_binary-0.2.0.post1.dist-info/METADATA +595 -0
  48. gllm_memory_binary-0.2.0.post1.dist-info/RECORD +50 -0
  49. gllm_memory_binary-0.2.0.post1.dist-info/WHEEL +5 -0
  50. gllm_memory_binary-0.2.0.post1.dist-info/top_level.txt +1 -0
@@ -0,0 +1,8 @@
1
+ from gllm_memory.builder.build_memory_client import build_memory_client as build_memory_client
2
+ from gllm_memory.builder.memory_manager_builder import MemoryManagerConfig as MemoryManagerConfig
3
+ from gllm_memory.clients.mem0.http.mem0_client import Mem0Client as Mem0Client
4
+ from gllm_memory.clients.memory_client import BaseMemoryClient as BaseMemoryClient
5
+ from gllm_memory.enums import MemoryProviderType as MemoryProviderType
6
+ from gllm_memory.memory_manager import MemoryManager as MemoryManager
7
+
8
+ __all__ = ['MemoryManager', 'MemoryManagerConfig', 'BaseMemoryClient', 'Mem0Client', 'MemoryProviderType', 'build_memory_client']
@@ -0,0 +1,5 @@
1
+ from gllm_memory.backends.base import BaseMemoryBackendAdapter as BaseMemoryBackendAdapter
2
+ from gllm_memory.backends.mem0 import Mem0BackendAdapter as Mem0BackendAdapter
3
+ from gllm_memory.backends.resolver import build_configured_memory_client as build_configured_memory_client, resolve_config_backend_adapter as resolve_config_backend_adapter
4
+
5
+ __all__ = ['BaseMemoryBackendAdapter', 'Mem0BackendAdapter', 'build_configured_memory_client', 'resolve_config_backend_adapter']
@@ -0,0 +1,34 @@
1
+ from abc import ABC, abstractmethod
2
+ from gllm_memory.builder.memory_manager_builder import MemoryManagerConfig as MemoryManagerConfig
3
+ from gllm_memory.clients.memory_client import BaseMemoryClient as BaseMemoryClient
4
+ from typing import Any
5
+
6
+ class BaseMemoryBackendAdapter(ABC):
7
+ """Abstract adapter interface for one internal memory backend.
8
+
9
+ Attributes:
10
+ backend_key (str): Stable internal identifier for the backend implementation.
11
+ """
12
+ backend_key: str
13
+ @abstractmethod
14
+ def can_handle_config(self, config: dict[str, Any] | MemoryManagerConfig) -> bool:
15
+ """Determine whether this adapter can handle one config object or payload.
16
+
17
+ Args:
18
+ config (dict[str, Any] | MemoryManagerConfig): Memory configuration candidate.
19
+
20
+ Returns:
21
+ bool: ``True`` if this adapter can build a runtime for the given config.
22
+ """
23
+ @abstractmethod
24
+ def build_configured_client(self, *, config: dict[str, Any] | MemoryManagerConfig, instruction: str | None = None, timeout_sec: int = 30) -> BaseMemoryClient:
25
+ """Build one configured memory client for this backend.
26
+
27
+ Args:
28
+ config (dict[str, Any] | MemoryManagerConfig): Memory configuration candidate.
29
+ instruction (str | None, optional): Optional instruction override. Defaults to None.
30
+ timeout_sec (int, optional): Request timeout in seconds. Defaults to 30.
31
+
32
+ Returns:
33
+ BaseMemoryClient: Concrete client instance for the selected backend.
34
+ """
@@ -0,0 +1,3 @@
1
+ from gllm_memory.backends.mem0.adapter import Mem0BackendAdapter as Mem0BackendAdapter
2
+
3
+ __all__ = ['Mem0BackendAdapter']
@@ -0,0 +1,35 @@
1
+ from _typeshed import Incomplete
2
+ from gllm_memory.backends.base import BaseMemoryBackendAdapter as BaseMemoryBackendAdapter
3
+ from gllm_memory.backends.mem0.config_compiler import PUBLIC_MEMORY_CONFIG_SECTIONS as PUBLIC_MEMORY_CONFIG_SECTIONS, compile_public_payload_to_mem0_config as compile_public_payload_to_mem0_config, looks_like_mem0_native_config as looks_like_mem0_native_config
4
+ from gllm_memory.builder.memory_manager_builder import MemoryManagerConfig as MemoryManagerConfig, clone_memory_config_payload as clone_memory_config_payload
5
+ from gllm_memory.clients.mem0.sdk.mem0_sdk_client import Mem0SDKClient as Mem0SDKClient
6
+ from gllm_memory.clients.memory_client import BaseMemoryClient as BaseMemoryClient
7
+ from gllm_memory.enums import MemoryProviderType as MemoryProviderType
8
+ from typing import Any
9
+
10
+ class Mem0BackendAdapter(BaseMemoryBackendAdapter):
11
+ """Backend adapter that compiles config payloads into Mem0 runtime clients."""
12
+ backend_key: Incomplete
13
+ def can_handle_config(self, config: dict[str, Any] | MemoryManagerConfig) -> bool:
14
+ """Determine whether this adapter can handle one config object or payload.
15
+
16
+ Args:
17
+ config (dict[str, Any] | MemoryManagerConfig): Memory configuration candidate.
18
+
19
+ Returns:
20
+ bool: ``True`` when the config is compatible with the Mem0 backend.
21
+ """
22
+ def build_configured_client(self, *, config: dict[str, Any] | MemoryManagerConfig, instruction: str | None = None, timeout_sec: int = 30) -> BaseMemoryClient:
23
+ """Build one Mem0 SDK client from the given config input.
24
+
25
+ Args:
26
+ config (dict[str, Any] | MemoryManagerConfig): Memory configuration candidate.
27
+ instruction (str | None, optional): Optional instruction override. Defaults to None.
28
+ timeout_sec (int, optional): Request timeout in seconds. Defaults to 30.
29
+
30
+ Returns:
31
+ BaseMemoryClient: Concrete ``Mem0SDKClient`` instance.
32
+
33
+ Raises:
34
+ ValueError: If the config is unsupported by this adapter.
35
+ """
@@ -0,0 +1,38 @@
1
+ from _typeshed import Incomplete
2
+ from gllm_memory.backends.mem0.defaults import DEFAULT_MEM0_BACKEND_OPTIONS as DEFAULT_MEM0_BACKEND_OPTIONS
3
+ from gllm_memory.builder.memory_manager_builder import clone_memory_config_payload as clone_memory_config_payload
4
+ from typing import Any
5
+
6
+ VERSION_KEY: str
7
+ VECTOR_STORE_KEY: str
8
+ EMBEDDER_KEY: str
9
+ LLM_KEY: str
10
+ RERANKER_KEY: str
11
+ PUBLIC_MEMORY_STORE_KEY: str
12
+ PUBLIC_EMBEDDING_KEY: str
13
+ HISTORY_DB_PATH_KEY: str
14
+ GRAPH_STORE_KEY: str
15
+ DISABLE_GRAPH_KEY: str
16
+ BACKEND_DISABLE_GRAPH_KEY: str
17
+ PUBLIC_MEMORY_CONFIG_SECTIONS: Incomplete
18
+ MEM0_NATIVE_KEYS: Incomplete
19
+
20
+ def looks_like_mem0_native_config(config: dict[str, Any]) -> bool:
21
+ """Determine whether one dict already looks like a Mem0-native SDK config.
22
+
23
+ Args:
24
+ config (dict[str, Any]): Candidate config dict.
25
+
26
+ Returns:
27
+ bool: ``True`` if the dict already contains Mem0-native root keys.
28
+ """
29
+ def compile_public_payload_to_mem0_config(payload: dict[str, Any], backend_options: dict[str, Any] | None = None) -> dict[str, Any]:
30
+ """Compile public config payload into Mem0-native SDK config.
31
+
32
+ Args:
33
+ payload (dict[str, Any]): Public config payload.
34
+ backend_options (dict[str, Any] | None, optional): Internal backend options. Defaults to None.
35
+
36
+ Returns:
37
+ dict[str, Any]: Mem0-native config dict.
38
+ """
@@ -0,0 +1,21 @@
1
+ from _typeshed import Incomplete
2
+ from dataclasses import dataclass
3
+ from typing import Any
4
+
5
+ MEM0_VERSION_KEY: str
6
+ MEM0_HISTORY_DB_PATH_KEY: str
7
+ MEM0_BACKEND_DISABLE_GRAPH_KEY: str
8
+
9
+ @dataclass(frozen=True, slots=True)
10
+ class Mem0BackendDefaults:
11
+ """Default root options for Mem0 SDK-backed configured memory."""
12
+ version: str = ...
13
+ history_db_path: str = ...
14
+ disable_graph: bool = ...
15
+ def to_backend_options(self) -> dict[str, Any]:
16
+ """Return a new backend-options payload using Mem0 option names."""
17
+
18
+ DEFAULT_MEM0_BACKEND_OPTIONS: Incomplete
19
+
20
+ def build_default_mem0_backend_options() -> dict[str, Any]:
21
+ """Build a mutable backend-options dict for one ``MemoryManagerConfigBuilder``."""
@@ -0,0 +1,29 @@
1
+ from gllm_memory.backends.base import BaseMemoryBackendAdapter as BaseMemoryBackendAdapter
2
+ from gllm_memory.backends.mem0 import Mem0BackendAdapter as Mem0BackendAdapter
3
+ from gllm_memory.builder.memory_manager_builder import MemoryManagerConfig as MemoryManagerConfig
4
+ from gllm_memory.clients.memory_client import BaseMemoryClient as BaseMemoryClient
5
+ from typing import Any
6
+
7
+ def resolve_config_backend_adapter(config: dict[str, Any] | MemoryManagerConfig) -> BaseMemoryBackendAdapter:
8
+ """Resolve one backend adapter for the given config object or payload.
9
+
10
+ Args:
11
+ config (dict[str, Any] | MemoryManagerConfig): Memory configuration candidate.
12
+
13
+ Returns:
14
+ BaseMemoryBackendAdapter: Matching backend adapter.
15
+
16
+ Raises:
17
+ ValueError: If no registered backend adapter can handle the config.
18
+ """
19
+ def build_configured_memory_client(*, config: dict[str, Any] | MemoryManagerConfig, instruction: str | None = None, timeout_sec: int = 30) -> BaseMemoryClient:
20
+ """Build one configured memory client using the resolved backend adapter.
21
+
22
+ Args:
23
+ config (dict[str, Any] | MemoryManagerConfig): Memory configuration candidate.
24
+ instruction (str | None, optional): Optional instruction override. Defaults to None.
25
+ timeout_sec (int, optional): Request timeout in seconds. Defaults to 30.
26
+
27
+ Returns:
28
+ BaseMemoryClient: Concrete client instance created by the selected adapter.
29
+ """
@@ -0,0 +1,4 @@
1
+ from gllm_memory.builder.build_memory_client import build_from_env as build_from_env, build_memory_client as build_memory_client
2
+ from gllm_memory.builder.memory_manager_builder import MemoryManagerConfig as MemoryManagerConfig, MemoryManagerConfigBuilder as MemoryManagerConfigBuilder
3
+
4
+ __all__ = ['build_from_env', 'build_memory_client', 'MemoryManagerConfig', 'MemoryManagerConfigBuilder']
@@ -0,0 +1,45 @@
1
+ from gllm_memory.backends.resolver import build_configured_memory_client as build_configured_memory_client
2
+ from gllm_memory.builder.memory_manager_builder import MemoryManagerConfig as MemoryManagerConfig
3
+ from gllm_memory.clients.mem0.http.mem0_client import Mem0Client as Mem0Client
4
+ from gllm_memory.clients.memory_client import BaseMemoryClient as BaseMemoryClient
5
+ from gllm_memory.enums import MemoryProviderType as MemoryProviderType
6
+ from typing import Any
7
+
8
+ def build_memory_client(provider: str, **kwargs) -> BaseMemoryClient:
9
+ """Create a memory client for the specified provider.
10
+
11
+ Args:
12
+ provider (str): The name of the memory provider.
13
+ **kwargs: Additional keyword arguments passed to the client constructor.
14
+
15
+ Returns:
16
+ BaseMemoryClient: The created memory client instance.
17
+
18
+ Raises:
19
+ ValueError: If the provider is not supported.
20
+ """
21
+ def build_from_env(api_key: str | None = None, instruction: str | None = None, host: str | None = None, config: dict[str, Any] | MemoryManagerConfig | None = None) -> BaseMemoryClient:
22
+ """Create a memory client from configuration.
23
+
24
+ Routes to the appropriate client based on what's provided:
25
+ 1. If config provided -> resolve internal configured backend
26
+ 2. Else if api_key or host provided -> Mem0Client (existing)
27
+ 3. Else read from environment -> Mem0Client (existing)
28
+
29
+ Args:
30
+ api_key (str | None, optional): API key for authentication. If None, will be read
31
+ from MEM0_API_KEY environment variable. Defaults to None.
32
+ instruction (str | None, optional): Custom instructions for memory handling.
33
+ Defaults to None.
34
+ host (str | None, optional): Host for the memory client.
35
+ Defaults to None.
36
+ config (dict[str, Any] | MemoryManagerConfig | None, optional): Configured memory payload
37
+ or config object. If provided, resolves the configured backend path. Defaults to None.
38
+
39
+ Returns:
40
+ BaseMemoryClient: The created memory client instance.
41
+
42
+ Raises:
43
+ RuntimeError: If API key is not provided and MEM0_API_KEY environment variable is missing.
44
+ ValueError: If the provider specified in environment is not supported.
45
+ """
@@ -0,0 +1,275 @@
1
+ from _typeshed import Incomplete
2
+ from dataclasses import dataclass
3
+ from gllm_memory.enums import MemoryProviderType as MemoryProviderType
4
+ from typing import Any
5
+
6
+ @dataclass(frozen=True, slots=True)
7
+ class _BuilderDefaults:
8
+ """Default values used by ``MemoryManagerConfigBuilder``."""
9
+ openai_embedder_model: str = ...
10
+ elasticsearch_auto_create_index: bool = ...
11
+ elasticsearch_use_ssl: bool = ...
12
+ elasticsearch_verify_certs: bool = ...
13
+
14
+ @dataclass(frozen=True, slots=True)
15
+ class _ConfigKeys:
16
+ """Internal keys used in builder and Mem0-compatible config payloads."""
17
+ memory_store: str = ...
18
+ embedding: str = ...
19
+ llm: str = ...
20
+ reranker: str = ...
21
+ @property
22
+ def required_sections(self) -> tuple[str, str, str]:
23
+ """Return config sections required before building."""
24
+
25
+ @dataclass(frozen=True, slots=True)
26
+ class _ProviderNames:
27
+ """Provider names handled specially by the builder."""
28
+ elasticsearch: str = ...
29
+ gllm_inference: str = ...
30
+ llm_reranker: str = ...
31
+ openai: str = ...
32
+
33
+ logger: Incomplete
34
+
35
+ class MemoryManagerConfig:
36
+ """Public configuration object for ``MemoryManager``.
37
+
38
+ Attributes:
39
+ instruction (str | None): Optional instruction carried with the config object.
40
+ """
41
+ instruction: Incomplete
42
+ def __init__(self, config: dict[str, Any], instruction: str | None = None, backend_key: str = ..., backend_options: dict[str, Any] | None = None) -> None:
43
+ """Initializes a new instance of the ``MemoryManagerConfig`` class.
44
+
45
+ Args:
46
+ config (dict[str, Any]): Underlying memory-engine config payload.
47
+ instruction (str | None, optional): Optional instruction stored with the config object.
48
+ Defaults to None.
49
+ backend_key (str, optional): Internal backend key selected by the library.
50
+ Defaults to ``MemoryProviderType.MEM0``.
51
+ backend_options (dict[str, Any] | None, optional): Internal backend-specific options.
52
+ Defaults to None.
53
+ """
54
+ @classmethod
55
+ def builder(cls) -> MemoryManagerConfigBuilder:
56
+ """Create a builder for ``MemoryManagerConfig``."""
57
+ def to_dict(self) -> dict[str, Any]:
58
+ """Return a safe copy of the underlying config payload."""
59
+ def get_backend_key(self) -> str:
60
+ """Return the internal backend key selected for this config object."""
61
+ def get_backend_options(self) -> dict[str, Any]:
62
+ """Return a safe copy of the internal backend-specific options."""
63
+
64
+ class MemoryManagerConfigBuilder:
65
+ """Builder for ``MemoryManagerConfig``.
66
+
67
+ Attributes:
68
+ memory_store (_MemoryManagerStoreBuilder): Memory-store configuration helper.
69
+ embedding (_MemoryManagerEmbeddingBuilder): Embedding configuration helper.
70
+ llm (_MemoryManagerLLMBuilder): LLM registration helper.
71
+ reranker (_MemoryManagerRerankerBuilder): Optional reranker configuration helper.
72
+ """
73
+ memory_store: Incomplete
74
+ embedding: Incomplete
75
+ llm: Incomplete
76
+ reranker: Incomplete
77
+ def __init__(self) -> None:
78
+ """Initializes a new instance of the ``MemoryManagerConfigBuilder`` class."""
79
+ def instruction(self, instruction: str | None) -> MemoryManagerConfigBuilder:
80
+ """Set one optional instruction stored with the config object.
81
+
82
+ Args:
83
+ instruction (str | None): Instruction text. ``None`` clears the stored instruction.
84
+
85
+ Returns:
86
+ MemoryManagerConfigBuilder: The builder itself.
87
+ """
88
+ def conversation_store_path(self, conversation_store_path: str) -> MemoryManagerConfigBuilder:
89
+ """Set the path used for local conversation-state persistence.
90
+
91
+ Args:
92
+ conversation_store_path (str): Local conversation-store path.
93
+
94
+ Returns:
95
+ MemoryManagerConfigBuilder: The builder itself.
96
+
97
+ Raises:
98
+ ValueError: If the path is blank, contains traversal, or cannot be created.
99
+ """
100
+ def build_config(self) -> dict[str, Any]:
101
+ """Build validated memory-engine config.
102
+
103
+ Returns:
104
+ dict[str, Any]: Deep-copied config dictionary.
105
+
106
+ Raises:
107
+ ValueError: If required sections are missing.
108
+ """
109
+ def build(self) -> MemoryManagerConfig:
110
+ """Build one ``MemoryManagerConfig`` object from the assembled config.
111
+
112
+ Returns:
113
+ MemoryManagerConfig: Built config object.
114
+ """
115
+ def set_provider_config(self, section: str, provider: str, config: dict[str, Any]) -> MemoryManagerConfigBuilder:
116
+ """Set one provider section in the underlying config.
117
+
118
+ Used by nested builders (``memory_store``, ``embedding``, ``llm``) in this module.
119
+
120
+ Args:
121
+ section (str): Section name such as ``memory_store`` or ``embedding``.
122
+ provider (str): Provider identifier.
123
+ config (dict[str, Any]): Provider config payload.
124
+
125
+ Returns:
126
+ MemoryManagerConfigBuilder: The builder itself.
127
+ """
128
+
129
+ class _MemoryManagerStoreBuilder:
130
+ """Memory-store builder for ``MemoryManagerConfigBuilder``."""
131
+ def __init__(self, root_builder: MemoryManagerConfigBuilder) -> None:
132
+ """Initializes a new instance of the ``_MemoryManagerStoreBuilder`` class.
133
+
134
+ Args:
135
+ root_builder (MemoryManagerConfigBuilder): Root builder.
136
+ """
137
+ def elasticsearch(self, *, host: str, port: int, collection_name: str, embedding_model_dims: int, user: str | None = None, password: str | None = None, auto_create_index: bool = ..., use_ssl: bool = ..., verify_certs: bool = ...) -> MemoryManagerConfigBuilder:
138
+ """Configure Elasticsearch-backed memory storage.
139
+
140
+ Args:
141
+ host (str): Elasticsearch host.
142
+ port (int): Elasticsearch port.
143
+ collection_name (str): Collection or index name.
144
+ embedding_model_dims (int): Embedding dimension count.
145
+ user (str | None, optional): Elasticsearch username. Defaults to None.
146
+ password (str | None, optional): Elasticsearch password. Defaults to None.
147
+ auto_create_index (bool, optional): Whether to auto-create index. Defaults to True.
148
+ use_ssl (bool, optional): Whether to use SSL. Defaults to False.
149
+ verify_certs (bool, optional): Whether to verify certificates. Defaults to False.
150
+
151
+ Returns:
152
+ MemoryManagerConfigBuilder: Root builder.
153
+ """
154
+ def register(self, provider: str, config: dict[str, Any] | None = None) -> MemoryManagerConfigBuilder:
155
+ """Register one generic memory-store provider without provider-specific defaults.
156
+
157
+ Args:
158
+ provider (str): Memory-store provider name passed through to the backend.
159
+ config (dict[str, Any] | None, optional): Provider config payload. Defaults to None.
160
+
161
+ Returns:
162
+ MemoryManagerConfigBuilder: Root builder.
163
+
164
+ Raises:
165
+ ValueError: If ``provider`` is empty.
166
+ """
167
+
168
+ class _MemoryManagerEmbeddingBuilder:
169
+ """Embedding builder for ``MemoryManagerConfigBuilder``."""
170
+ def __init__(self, root_builder: MemoryManagerConfigBuilder) -> None:
171
+ """Initializes a new instance of the ``_MemoryManagerEmbeddingBuilder`` class.
172
+
173
+ Args:
174
+ root_builder (MemoryManagerConfigBuilder): Root builder.
175
+ """
176
+ def openai(self, *, model: str = ..., api_key: str | None = None) -> MemoryManagerConfigBuilder:
177
+ """Configure OpenAI embedding provider.
178
+
179
+ Args:
180
+ model (str, optional): Embedding model name. Defaults to ``text-embedding-3-small``.
181
+ api_key (str | None, optional): OpenAI API key. Defaults to None.
182
+
183
+ Returns:
184
+ MemoryManagerConfigBuilder: Root builder.
185
+ """
186
+ def register(self, em_invoker: Any, *, model: str | None = None, embedding_dims: int | None = None, invoke_kwargs: dict[str, Any] | None = None) -> MemoryManagerConfigBuilder:
187
+ """Register ``gllm_inference`` EM Invoker as the memory embedding backend.
188
+
189
+ Args:
190
+ em_invoker (Any): Configured EM Invoker instance.
191
+ model (str | None, optional): Optional model label stored in Mem0 config. Defaults to None.
192
+ embedding_dims (int | None, optional): Optional embedding dimension metadata. Defaults to None.
193
+ invoke_kwargs (dict[str, Any] | None, optional): Extra ``invoke`` kwargs. Defaults to None.
194
+
195
+ Returns:
196
+ MemoryManagerConfigBuilder: Root builder.
197
+
198
+ Raises:
199
+ ValueError: If ``em_invoker`` is ``None``.
200
+ """
201
+
202
+ class _MemoryManagerLLMBuilder:
203
+ """LLM builder for ``MemoryManagerConfigBuilder``."""
204
+ def __init__(self, root_builder: MemoryManagerConfigBuilder) -> None:
205
+ """Initializes a new instance of the ``_MemoryManagerLLMBuilder`` class.
206
+
207
+ Args:
208
+ root_builder (MemoryManagerConfigBuilder): Root builder.
209
+ """
210
+ def register(self, lm_invoker: Any, *, model: str | None = None, temperature: float = 0.1, invoke_kwargs: dict[str, Any] | None = None) -> MemoryManagerConfigBuilder:
211
+ """Register ``gllm_inference`` LM Invoker as the memory LLM backend.
212
+
213
+ Args:
214
+ lm_invoker (Any): Configured LM Invoker instance.
215
+ model (str | None, optional): Optional model label stored in Mem0 config. Defaults to None.
216
+ temperature (float, optional): Temperature forwarded to adapter config. Defaults to 0.1.
217
+ invoke_kwargs (dict[str, Any] | None, optional): Extra ``invoke`` kwargs. Defaults to None.
218
+
219
+ Returns:
220
+ MemoryManagerConfigBuilder: Root builder.
221
+
222
+ Raises:
223
+ ValueError: If ``lm_invoker`` is ``None``.
224
+ """
225
+
226
+ class _MemoryManagerRerankerBuilder:
227
+ """Reranker builder for ``MemoryManagerConfigBuilder``."""
228
+ def __init__(self, root_builder: MemoryManagerConfigBuilder) -> None:
229
+ """Initializes a new instance of the ``_MemoryManagerRerankerBuilder`` class.
230
+
231
+ Args:
232
+ root_builder (MemoryManagerConfigBuilder): Root builder.
233
+ """
234
+ def llm_reranker(self, *, llm_provider: str = ..., model: str | None = None, api_key: str | None = None, top_k: int | None = None, temperature: float = 0.0, max_tokens: int = 100, scoring_prompt: str | None = None, llm_config: dict[str, Any] | None = None) -> MemoryManagerConfigBuilder:
235
+ """Configure an LLM-based reranker.
236
+
237
+ Args:
238
+ llm_provider (str, optional): LLM provider used by the reranker. Defaults to ``openai``.
239
+ model (str | None, optional): Reranker LLM model name. Defaults to None.
240
+ api_key (str | None, optional): Provider API key. Defaults to None.
241
+ top_k (int | None, optional): Number of memories to keep after reranking. Defaults to None.
242
+ temperature (float, optional): Reranker LLM temperature. Defaults to 0.0.
243
+ max_tokens (int, optional): Maximum reranker LLM response tokens. Defaults to 100.
244
+ scoring_prompt (str | None, optional): Custom scoring prompt. Defaults to None.
245
+ llm_config (dict[str, Any] | None, optional): Extra nested LLM config. Defaults to None.
246
+
247
+ Returns:
248
+ MemoryManagerConfigBuilder: Root builder.
249
+
250
+ Raises:
251
+ ValueError: If ``llm_provider`` is empty.
252
+ """
253
+ def register(self, provider: str, config: dict[str, Any] | None = None) -> MemoryManagerConfigBuilder:
254
+ """Register one generic reranker provider without provider-specific defaults.
255
+
256
+ Args:
257
+ provider (str): Reranker provider name passed through to the backend.
258
+ config (dict[str, Any] | None, optional): Provider config payload. Defaults to None.
259
+
260
+ Returns:
261
+ MemoryManagerConfigBuilder: Root builder.
262
+
263
+ Raises:
264
+ ValueError: If ``provider`` is empty.
265
+ """
266
+
267
+ def clone_memory_config_payload(config: dict[str, Any]) -> dict[str, Any]:
268
+ """Clone one config payload while preserving non-copyable runtime objects.
269
+
270
+ Args:
271
+ config (dict[str, Any]): Builder config.
272
+
273
+ Returns:
274
+ dict[str, Any]: Cloned config.
275
+ """
@@ -0,0 +1,99 @@
1
+ from _typeshed import Incomplete
2
+ from gllm_memory.backends.mem0.defaults import MEM0_HISTORY_DB_PATH_KEY as MEM0_HISTORY_DB_PATH_KEY, build_default_mem0_backend_options as build_default_mem0_backend_options
3
+ from typing import Any
4
+
5
+ DEFAULT_OPENAI_EMBEDDER_MODEL: str
6
+ DEFAULT_OPENAI_LLM_MODEL: str
7
+ DEFAULT_ELASTICSEARCH_AUTO_CREATE_INDEX: bool
8
+ DEFAULT_ELASTICSEARCH_USE_SSL: bool
9
+ DEFAULT_ELASTICSEARCH_VERIFY_CERTS: bool
10
+ CONVERSATION_STORE_PATH_BACKEND_OPTION_KEY = MEM0_HISTORY_DB_PATH_KEY
11
+ ELASTICSEARCH_HOST_ENV: str
12
+ ELASTICSEARCH_PORT_ENV: str
13
+ ELASTICSEARCH_USER_ENV: str
14
+ ELASTICSEARCH_PASSWORD_ENV: str
15
+ ELASTICSEARCH_COLLECTION_ENV: str
16
+ ELASTICSEARCH_EMBEDDING_DIMS_ENV: str
17
+ OPENAI_API_KEY_ENV: str
18
+ OPENAI_BASE_URL_ENV: str
19
+ OPENAI_MODEL_NAME_ENV: str
20
+ logger: Incomplete
21
+
22
+ def build_default_backend_options() -> dict[str, Any]:
23
+ """Build default backend options for the configured memory runtime.
24
+
25
+ Returns:
26
+ dict[str, Any]: Backend options for ``MemoryManagerConfig``.
27
+ """
28
+ def get_required_env(name: str) -> str:
29
+ """Get one required environment variable or raise with an error log.
30
+
31
+ Args:
32
+ name (str): Environment variable name.
33
+
34
+ Returns:
35
+ str: Environment variable value.
36
+
37
+ Raises:
38
+ ValueError: If the environment variable is missing or empty.
39
+ """
40
+ def get_optional_env(name: str) -> str | None:
41
+ """Get one optional environment variable.
42
+
43
+ Args:
44
+ name (str): Environment variable name.
45
+
46
+ Returns:
47
+ str | None: Environment variable value if present, else ``None``.
48
+ """
49
+ def get_required_int_env(name: str) -> int:
50
+ """Get one required integer environment variable.
51
+
52
+ Args:
53
+ name (str): Environment variable name.
54
+
55
+ Returns:
56
+ int: Parsed integer value.
57
+
58
+ Raises:
59
+ ValueError: If the variable is missing or not an integer.
60
+ """
61
+ def get_optional_int_env(name: str) -> int | None:
62
+ """Get one optional integer environment variable.
63
+
64
+ Args:
65
+ name (str): Environment variable name.
66
+
67
+ Returns:
68
+ int | None: Parsed integer value if present, else ``None``.
69
+
70
+ Raises:
71
+ ValueError: If the variable is set but not parseable as ``int``.
72
+ """
73
+ def build_default_elasticsearch_config() -> dict[str, Any]:
74
+ """Build default Elasticsearch memory-store config from environment variables.
75
+
76
+ Returns:
77
+ dict[str, Any]: Elasticsearch provider config payload.
78
+
79
+ Raises:
80
+ ValueError: If required environment variables are missing or invalid.
81
+ """
82
+ def build_default_openai_embedding_config() -> dict[str, Any]:
83
+ """Build default OpenAI EM Invoker config from environment variables.
84
+
85
+ Returns:
86
+ dict[str, Any]: ``gllm_inference`` embedding provider config payload.
87
+
88
+ Raises:
89
+ ValueError: If required environment variables are missing.
90
+ """
91
+ def build_default_openai_llm_config() -> dict[str, Any]:
92
+ """Build default OpenAI LM Invoker config from environment variables.
93
+
94
+ Returns:
95
+ dict[str, Any]: ``gllm_inference`` provider config payload.
96
+
97
+ Raises:
98
+ ValueError: If required environment variables are missing.
99
+ """
@@ -0,0 +1 @@
1
+ from gllm_memory.chat_history_manager.chat_history_manager import ChatHistoryManager as ChatHistoryManager
@@ -0,0 +1,3 @@
1
+ from gllm_memory.clients.constants import DATETIME_FULL_LENGTH as DATETIME_FULL_LENGTH, DATETIME_NO_SEC_LENGTH as DATETIME_NO_SEC_LENGTH, DATE_ONLY_LENGTH as DATE_ONLY_LENGTH, PEPPER as PEPPER, TIME_ONLY_LENGTH as TIME_ONLY_LENGTH
2
+
3
+ __all__ = ['DATE_ONLY_LENGTH', 'DATETIME_FULL_LENGTH', 'DATETIME_NO_SEC_LENGTH', 'PEPPER', 'TIME_ONLY_LENGTH']
@@ -0,0 +1,7 @@
1
+ __all__ = ['DATE_ONLY_LENGTH', 'DATETIME_FULL_LENGTH', 'DATETIME_NO_SEC_LENGTH', 'PEPPER', 'TIME_ONLY_LENGTH']
2
+
3
+ PEPPER: str
4
+ DATETIME_FULL_LENGTH: int
5
+ DATETIME_NO_SEC_LENGTH: int
6
+ DATE_ONLY_LENGTH: int
7
+ TIME_ONLY_LENGTH: int
@@ -0,0 +1,5 @@
1
+ from gllm_memory.clients.mem0.core.mem0_base_client import Mem0BaseClient as Mem0BaseClient
2
+ from gllm_memory.clients.mem0.http.mem0_client import Mem0Client as Mem0Client
3
+ from gllm_memory.clients.mem0.sdk.mem0_sdk_client import DEFAULT_MIN_PAYLOADS_FOR_PARALLEL_PARSE as DEFAULT_MIN_PAYLOADS_FOR_PARALLEL_PARSE, Mem0SDKClient as Mem0SDKClient
4
+
5
+ __all__ = ['DEFAULT_MIN_PAYLOADS_FOR_PARALLEL_PARSE', 'Mem0BaseClient', 'Mem0Client', 'Mem0SDKClient']
@@ -0,0 +1,3 @@
1
+ from gllm_memory.clients.mem0.core.mem0_base_client import Mem0BaseClient as Mem0BaseClient
2
+
3
+ __all__ = ['Mem0BaseClient']