DeepFabric 4.4.0__py3-none-any.whl → 4.4.1__py3-none-any.whl

deepfabric/llm/client.py

@@ -1,5 +1,7 @@
 import asyncio
+import logging
 import os
+import sys
 
 from functools import lru_cache
 from typing import Any
@@ -21,6 +23,8 @@ from .rate_limit_config import (
 )
 from .retry_handler import RetryHandler, retry_with_backoff, retry_with_backoff_async
 
+logger = logging.getLogger(__name__)
+
 # JSON Schema union type keys that need recursive processing
 _UNION_KEYS = ("anyOf", "oneOf", "allOf")
 
@@ -1061,8 +1065,18 @@ def _get_cached_openai_schema(schema: type[BaseModel]) -> type[BaseModel]:
     OpenAICompatModel.__name__ = f"{schema.__name__}OpenAICompat"
     OpenAICompatModel.__doc__ = schema.__doc__
 
-    # Rebuild model to resolve forward references (e.g., PendingToolCall in AgentStep)
-    OpenAICompatModel.model_rebuild()
+    # Rebuild model with the schema's original module namespace to resolve
+    # forward references (e.g., PendingToolCall in AgentStep)
+    schema_module = sys.modules.get(schema.__module__)
+    if schema_module:
+        OpenAICompatModel.model_rebuild(_types_namespace=vars(schema_module))
+    else:
+        logger.warning(
+            "Could not find module '%s' in sys.modules. "
+            "Forward reference resolution for dynamically created models may fail.",
+            schema.__module__,
+        )
+        OpenAICompatModel.model_rebuild()
 
     return OpenAICompatModel
 

{deepfabric-4.4.0.dist-info → deepfabric-4.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: DeepFabric
-Version: 4.4.0
+Version: 4.4.1
 Summary: Curate High Quality Datasets, Train, Evaluate and Ship
 Author-email: Luke Hinds <luke@alwaysfurther.ai>
 License-File: LICENSE

{deepfabric-4.4.0.dist-info → deepfabric-4.4.1.dist-info}/RECORD

@@ -48,7 +48,7 @@ deepfabric/evaluation/reporters/file_reporter.py,sha256=cRSrtFjYdrUoSwCHPLd6e5ir
 deepfabric/evaluation/reporters/multi_reporter.py,sha256=Faxi_cgLIVsRzti9n6zd_fxWUToSam65Rqylp6RX97c,1621
 deepfabric/llm/__init__.py,sha256=lfWDLLQ6VWJ4birKVRFh3ypk9zEOfG1ZtU08iD8bd3U,1038
 deepfabric/llm/api_key_verifier.py,sha256=pxkijmXH3fq23p53xerwEARHmTaz4FnphN-FTU-i42Q,17347
-deepfabric/llm/client.py,sha256=mkW6sElY6CdUbLttO6xAdG1LRaHQ4_SAC36vvn05ioI,46297
+deepfabric/llm/client.py,sha256=7__p0KsN9y8wbqpnlHnVOe92quoQs7K2qd_mrU-l2w0,46788
 deepfabric/llm/errors.py,sha256=b25V4iJHkdAgem2ufJkPuornkesyVgQVPQHWttigB54,4293
 deepfabric/llm/rate_limit_config.py,sha256=-skBR8PbEMzsPGeAaiYYFuHAY48AGkl7DIVklijUI-c,8339
 deepfabric/llm/rate_limit_detector.py,sha256=hbp35snCAkn7uBsPaeq7-8XNBngFH1ecS3scRSKP_y4,9960
@@ -64,8 +64,8 @@ deepfabric/training/__init__.py,sha256=MrfbMhPUa_h4zqpTWDiIu_dXKhmlUH1tJsO5mIsm6
 deepfabric/training/api_key_prompt.py,sha256=bzcdzeK6ql_8Vz0cbR2vmxxtMRNRFpzYAJx7i8GNI3U,9315
 deepfabric/training/callback.py,sha256=d4heAO5gcBCGXmJIgA9MI9lFBBHnYtdm76rhPh6RDSA,11427
 deepfabric/training/metrics_sender.py,sha256=Fh_qvqrK9mNpTHfG8jgU7t1oSMKbov3mmj3r6o_Q6X4,9347
-deepfabric-4.4.0.dist-info/METADATA,sha256=SYHJO9vj_Rc480ZdhDEbAAM403jjN-RYaX4nBYRX1Tw,22606
-deepfabric-4.4.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-deepfabric-4.4.0.dist-info/entry_points.txt,sha256=zatevils13hfs8x29_vmUyivQ6rTtq7hE2RBusZw1Fo,50
-deepfabric-4.4.0.dist-info/licenses/LICENSE,sha256=-qRt8wmrhQ9aMf7KhmZXc2vrTETYZF-6_T1KCeUhvHY,11340
-deepfabric-4.4.0.dist-info/RECORD,,
+deepfabric-4.4.1.dist-info/METADATA,sha256=EGTNsfWuaMrduG33KLktG7mLyVF73HgYw1y3VJEKN2g,22606
+deepfabric-4.4.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+deepfabric-4.4.1.dist-info/entry_points.txt,sha256=zatevils13hfs8x29_vmUyivQ6rTtq7hE2RBusZw1Fo,50
+deepfabric-4.4.1.dist-info/licenses/LICENSE,sha256=-qRt8wmrhQ9aMf7KhmZXc2vrTETYZF-6_T1KCeUhvHY,11340
+deepfabric-4.4.1.dist-info/RECORD,,