agent-lab-sdk 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

agent_lab_sdk/schema/__init__.py

@@ -0,0 +1,2 @@
+from log_message import LogMessage
+__all__ = ["LogMessage"]

agent_lab_sdk/schema/log_message.py

@@ -0,0 +1,17 @@
+from langchain.schema import BaseMessage, AIMessage
+import os
+
+IS_LOG_MESSAGE_CUSTOM = os.environ.get("IS_LOG_MESSAGE_CUSTOM", 'False').lower() in ('true', '1', 't')
+
+if IS_LOG_MESSAGE_CUSTOM:
+    class LogMessage(BaseMessage):
+        def __init__(self, content: str):
+            super().__init__(type='log', content=content, additional_kwargs={})
+        def dict(self, **kwargs):
+            d = super().dict(**kwargs)
+            d["type"] = 'log'
+            return d
+else:
+    class LogMessage(AIMessage):
+        def __init__(self, content: str):
+            super().__init__(content=content, additional_kwargs = {"type": "log"})

{agent_lab_sdk-0.1.7.dist-info → agent_lab_sdk-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-lab-sdk
-Version: 0.1.7
+Version: 0.1.9
 Summary: SDK для работы с Agent Lab
 Author-email: Andrew Ohurtsov <andermirik@yandex.com>
 License: Proprietary and Confidential — All Rights Reserved

{agent_lab_sdk-0.1.7.dist-info → agent_lab_sdk-0.1.9.dist-info}/RECORD

@@ -8,10 +8,12 @@ agent_lab_sdk/llm/llm.py,sha256=NOEH9TOH66EIJXGevxPm6w6px7Z0cZl9DJ-9A7jOnd0,873
 agent_lab_sdk/llm/throttled.py,sha256=9_nm1i3Uuep0VEWsY1KNCllZA-vM202XVdlgXhgC8BA,7005
 agent_lab_sdk/metrics/__init__.py,sha256=G4VSlzKwupPMM4c6vZaF1rnd0KusKarezDMjli9pVFw,57
 agent_lab_sdk/metrics/metrics.py,sha256=2e0c7BanThUNtCxpS6BUlAIDoLSidQsuaaBP5EB48Yo,3432
+agent_lab_sdk/schema/__init__.py,sha256=gUD5ymdagTp3bGDFPy90u39leZ0fOVVSJB5Ssa3VP_s,60
+agent_lab_sdk/schema/log_message.py,sha256=nadi6lZGRuDSPmfbYs9QPpRJUT9Pfy8Y7pGCvyFF5Mw,638
 agent_lab_sdk/storage/__init__.py,sha256=ik1_v1DMTwehvcAEXIYxuvLuCjJCa3y5qAuJqoQpuSA,81
 agent_lab_sdk/storage/storage.py,sha256=ELpt7GRwFD-aWa6ctinfA_QwcvzWLvKS0Wz8FlxVqAs,2075
-agent_lab_sdk-0.1.7.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
-agent_lab_sdk-0.1.7.dist-info/METADATA,sha256=Oa81Mo3xrEyfH7BZ8rXV03MyNzlj3mysFBpOIUQID58,1748
-agent_lab_sdk-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-agent_lab_sdk-0.1.7.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
-agent_lab_sdk-0.1.7.dist-info/RECORD,,
+agent_lab_sdk-0.1.9.dist-info/licenses/LICENSE,sha256=_TRXHkF3S9ilWBPdZcHLI_S-PRjK0L_SeOb2pcPAdV4,417
+agent_lab_sdk-0.1.9.dist-info/METADATA,sha256=OEaBn-h99uhnd7xcSvWoPl-wUv0zipvJglu3PyxHO5M,1748
+agent_lab_sdk-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+agent_lab_sdk-0.1.9.dist-info/top_level.txt,sha256=E1efqkJ89KNmPBWdLzdMHeVtH0dYyCo4fhnSb81_15I,14
+agent_lab_sdk-0.1.9.dist-info/RECORD,,