amsdal_ml 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.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.
amsdal_ml/__about__.py CHANGED
@@ -1 +1 @@
1
- __version__ = '0.1.1'
1
+ __version__ = '0.1.3'
amsdal_ml/app.py CHANGED
@@ -1,24 +1,8 @@
1
- from __future__ import annotations
1
+ from amsdal.contrib.app_config import AppConfig
2
2
 
3
- from typing import TYPE_CHECKING
4
3
 
5
- # During type checking, point to the known-available fallback.
6
- if TYPE_CHECKING:
7
- from amsdal.contrib.app_config import AppConfig as BaseAppConfig
8
- else:
9
- try:
10
- # At runtime prefer the real AMSDAL core if present.
11
- from amsdal.configs.app import AppConfig as BaseAppConfig # type: ignore[import-not-found]
12
- except Exception: # pragma: no cover
13
- from amsdal.contrib.app_config import AppConfig as BaseAppConfig
14
-
15
-
16
- class MLPluginAppConfig(BaseAppConfig):
4
+ class MLPluginAppConfig(AppConfig):
17
5
  name = "amsdal_ml"
18
6
  verbose_name = "AMSDAL ML Plugin"
19
7
 
20
- def on_ready(self) -> None:
21
- pass
22
-
23
- def on_server_startup(self) -> None:
24
- pass
8
+ def on_ready(self) -> None: ...
@@ -2,6 +2,7 @@ from __future__ import annotations
2
2
 
3
3
  import asyncio
4
4
  import base64
5
+ import os
5
6
  from collections.abc import Iterable
6
7
  from contextlib import AsyncExitStack
7
8
  from typing import Any
@@ -47,7 +48,7 @@ class StdioClient(ToolClient):
47
48
 
48
49
  async def _open_session(self) -> tuple[AsyncExitStack, ClientSession]:
49
50
  stack = AsyncExitStack()
50
- params = StdioServerParameters(command=self._command, args=self._args)
51
+ params = StdioServerParameters(command=self._command, args=self._args, env=os.environ.copy())
51
52
  rx, tx = await stack.enter_async_context(stdio_client(params))
52
53
  s = await stack.enter_async_context(ClientSession(rx, tx))
53
54
  await s.initialize()
@@ -9,10 +9,10 @@ class Migration(migrations.Migration):
9
9
  class_name="EmbeddingModel",
10
10
  new_schema={
11
11
  "title": "EmbeddingModel",
12
- "required": ["object_class", "object_id", "chunk_index", "raw_text", "embedding"],
12
+ "required": ["data_object_class", "data_object_id", "chunk_index", "raw_text", "embedding"],
13
13
  "properties": {
14
- "object_class": {"type": "string", "title": "Linked object class"},
15
- "object_id": {"type": "string", "title": "Linked object ID"},
14
+ "data_object_class": {"type": "string", "title": "Linked object class"},
15
+ "data_object_id": {"type": "string", "title": "Linked object ID"},
16
16
  "chunk_index": {"type": "integer", "title": "Chunk index"},
17
17
  "raw_text": {"type": "string", "title": "Raw text used for embedding"},
18
18
  "embedding": {
@@ -288,8 +288,8 @@ class DefaultIngesting(MLIngesting):
288
288
  object_class, object_id = self._resolve_link(instance)
289
289
  objs = [
290
290
  EmbeddingModel(
291
- object_class=object_class,
292
- object_id=object_id,
291
+ data_object_class=object_class,
292
+ data_object_id=object_id,
293
293
  chunk_index=r.chunk_index,
294
294
  raw_text=r.raw_text,
295
295
  embedding=r.embedding,
@@ -305,8 +305,8 @@ class DefaultIngesting(MLIngesting):
305
305
  object_class, object_id = self._resolve_link(instance)
306
306
  objs = [
307
307
  EmbeddingModel(
308
- object_class=object_class,
309
- object_id=object_id,
308
+ data_object_class=object_class,
309
+ data_object_id=object_id,
310
310
  chunk_index=r.chunk_index,
311
311
  raw_text=r.raw_text,
312
312
  embedding=r.embedding,
@@ -60,8 +60,8 @@ class DefaultRetriever(MLRetriever, ABC):
60
60
 
61
61
  return [
62
62
  RetrievalChunk(
63
- object_class=r.object_class,
64
- object_id=r.object_id,
63
+ object_class=r.data_object_class,
64
+ object_id=r.data_object_id,
65
65
  chunk_index=r.chunk_index,
66
66
  raw_text=(r.raw_text or '').strip(),
67
67
  distance=float(getattr(r, 'distance', math.inf)),
@@ -94,8 +94,8 @@ class DefaultRetriever(MLRetriever, ABC):
94
94
 
95
95
  return [
96
96
  RetrievalChunk(
97
- object_class=r.object_class,
98
- object_id=r.object_id,
97
+ object_class=r.data_object_class,
98
+ object_id=r.data_object_id,
99
99
  chunk_index=r.chunk_index,
100
100
  raw_text=(r.raw_text or '').strip(),
101
101
  distance=float(getattr(r, 'distance', math.inf)),
@@ -10,8 +10,8 @@ class EmbeddingModel(Model):
10
10
  __module_type__ = ModuleType.CONTRIB
11
11
  __table_name__ = "embedding_model"
12
12
 
13
- object_class: str = Field(..., title="Linked object class")
14
- object_id: str = Field(..., title="Linked object ID")
13
+ data_object_class: str = Field(..., title="Linked object class")
14
+ data_object_id: str = Field(..., title="Linked object ID")
15
15
 
16
16
  chunk_index: int = Field(..., title="Chunk index")
17
17
  raw_text: str = Field(..., title="Raw text used for embedding")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: amsdal_ml
3
- Version: 0.1.1
3
+ Version: 0.1.3
4
4
  Summary: amsdal_ml plugin for AMSDAL Framework
5
5
  Requires-Python: >=3.11
6
6
  Requires-Dist: aiohttp==3.12.15
@@ -1,7 +1,7 @@
1
1
  amsdal_ml/Third-Party Materials - AMSDAL Dependencies - License Notices.md,sha256=0A3jjw91I-NaTHXCw8IQiKUdFJgVavhgHaFX4Nag4jk,81658
2
- amsdal_ml/__about__.py,sha256=ls1camlIoMxEZz9gSkZ1OJo-MXqHWwKPtdPbZJmwp7E,22
2
+ amsdal_ml/__about__.py,sha256=uZsygMXMKRw-7qhWojAjnpm8GFPXU92xW6XA8O5GwFY,22
3
3
  amsdal_ml/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
- amsdal_ml/app.py,sha256=yq_XIstjXSKplsjHLn9Jb65zXx4HOdjPKVlbz5FT9ec,715
4
+ amsdal_ml/app.py,sha256=K_lUz37_FUQmPwzDakDhVXA_D2SL9z1JtcuHfuBvlKE,184
5
5
  amsdal_ml/ml_config.py,sha256=UONMVG2RbKVbaYF-GdlhRQbm8flp52GxBzfqdzuLZ5w,1788
6
6
  amsdal_ml/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
7
  amsdal_ml/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -16,12 +16,12 @@ amsdal_ml/fileio/openai_loader.py,sha256=4YyxK5pBAnRm55et5yMu6X5OdK2kWTqpuPF1E2-
16
16
  amsdal_ml/mcp_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
17
17
  amsdal_ml/mcp_client/base.py,sha256=cNuHHCSSND4BkIyhQZ-1mJLPp630wTCtiCQSHnK2zmA,441
18
18
  amsdal_ml/mcp_client/http_client.py,sha256=2fFCaubp_cFeyCqN9CP7bVLTp4UIIEhJPq9BVHBf6-w,1719
19
- amsdal_ml/mcp_client/stdio_client.py,sha256=bYfS2yCeJNl8fxBn6Yerg4RBZKbW2pgcdKiwX-NL6io,5744
19
+ amsdal_ml/mcp_client/stdio_client.py,sha256=F5Y2yGFNDCE7Nfg_6QMCkp81Vb4uS2UQ1XnD_FTGM8U,5777
20
20
  amsdal_ml/mcp_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
21
21
  amsdal_ml/mcp_server/server_retriever_stdio.py,sha256=iUQNwWEDZE0OIS9vyTyJp_EOn4kvl9Gn0Okzcq-ymo0,1762
22
- amsdal_ml/migrations/0000_initial.py,sha256=vbgDcBK3_CI0jUn0sl0e4_XdDncBHAQ7T4x2FZZkGnQ,1613
22
+ amsdal_ml/migrations/0000_initial.py,sha256=UeEtqHrCgmTQyGnyV2jC7yt9Ce7Xm0-MyhOQU3rRrs8,1633
23
23
  amsdal_ml/ml_ingesting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
24
- amsdal_ml/ml_ingesting/default_ingesting.py,sha256=Hg5D_CjIZJAxH8juw42R9i-OyUlL0qzhbn6R43ETkA4,12839
24
+ amsdal_ml/ml_ingesting/default_ingesting.py,sha256=udVAyUbjeCrsP9pk-VbAH689M3VmBRhOFbl9Que9oK0,12859
25
25
  amsdal_ml/ml_ingesting/embedding_data.py,sha256=ZWxhYq8mkFVC3HMM5nEiObgy62X8hLh1McaGegnRUsk,355
26
26
  amsdal_ml/ml_ingesting/ingesting.py,sha256=h-GdNCtutO8C75jbLko40CMjHym9Xk_nlMdNYLfBs58,1458
27
27
  amsdal_ml/ml_ingesting/openai_ingesting.py,sha256=_2dU2id2a3xR_y0GvZc5LUDw3_wI-wbhEFogEOKF-H0,1421
@@ -29,11 +29,11 @@ amsdal_ml/ml_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
29
29
  amsdal_ml/ml_models/models.py,sha256=gZzZrjsoAnFc5pPwqqLPNh5RYxwKKaFyNg56NJnsn4w,2286
30
30
  amsdal_ml/ml_models/openai_model.py,sha256=yQ6r4K7AOeG3NfWT6pXgjkRPMT0FuQPqFUWeHSwz0u8,14295
31
31
  amsdal_ml/ml_retrievers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
32
- amsdal_ml/ml_retrievers/default_retriever.py,sha256=ncB9jnvnEfw2j58mINn9vs34pfTHE838cs30SjGkt7Y,3213
32
+ amsdal_ml/ml_retrievers/default_retriever.py,sha256=ykAzi6gESyG1693nALpPJ9XhKV_We1yrNxbYo1eG748,3233
33
33
  amsdal_ml/ml_retrievers/openai_retriever.py,sha256=oY71oPJdFuECWaJL_ByKZFE7HRJ-S6ypRzzJ7XjI7H8,1236
34
34
  amsdal_ml/ml_retrievers/retriever.py,sha256=zHvaAOAD6H4IaIaYA_wy2e8tMmGtOnyNsDS2B4dQjBU,1005
35
35
  amsdal_ml/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
36
- amsdal_ml/models/embedding_model.py,sha256=26sLxJ1VPbW0ZuY0mQOfHm_ADcKNzxq9_ffJ91SzHg8,764
37
- amsdal_ml-0.1.1.dist-info/METADATA,sha256=doT-XMQ_fMnwVg2Wi1hwuIN_mUIVQYgRLYIBwZYmEDg,1676
38
- amsdal_ml-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
39
- amsdal_ml-0.1.1.dist-info/RECORD,,
36
+ amsdal_ml/models/embedding_model.py,sha256=mdMSYHrkSIKHi5mg4HqMaTTiPGqLYtRdRkbEK_trFWw,774
37
+ amsdal_ml-0.1.3.dist-info/METADATA,sha256=BETzPNugrsUz-RL3yG-2I-4I_WBSLkJ5oRJYxyZxEDY,1676
38
+ amsdal_ml-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
39
+ amsdal_ml-0.1.3.dist-info/RECORD,,