prefactor-langchain 0.2.2__tar.gz → 0.2.3__tar.gz

{prefactor_langchain-0.2.2 → prefactor_langchain-0.2.3}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: prefactor-langchain
-Version: 0.2.2
+Version: 0.2.3
 Summary: LangChain integration for Prefactor observability
 Author-email: Prefactor Pty Ltd <josh@prefactor.tech>
 License: MIT
 Requires-Python: <4.0.0,>=3.11.0
 Requires-Dist: langchain-core>=1.0.0
-Requires-Dist: prefactor-core>=0.2.1
+Requires-Dist: prefactor-core>=0.2.2
 Description-Content-Type: text/markdown
 
 # prefactor-langchain

{prefactor_langchain-0.2.2 → prefactor_langchain-0.2.3}/examples/simple_agent.py

@@ -50,7 +50,7 @@ def _safe_eval(node: ast.expr) -> float:
     if isinstance(node, ast.Constant):
         return node.n  # type: ignore[return-value]
     if isinstance(node, ast.BinOp):
-        return _OPS[type(node.op)](_safe_eval(node.left), _safe_eval(node.right))
+        return _OPS[type(node.op)](_safe_eval(node.left), _safe_eval(node.right))  # type: ignore[invalid-argument-type]
     if isinstance(node, ast.UnaryOp) and isinstance(node.op, ast.USub):
         return -_safe_eval(node.operand)
     raise ValueError(f"Unsupported expression: {ast.dump(node)}")

{prefactor_langchain-0.2.2 → prefactor_langchain-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "prefactor-langchain"
-version = "0.2.2"
+version = "0.2.3"
 description = "LangChain integration for Prefactor observability"
 readme = "README.md"
 license = { text = "MIT" }
@@ -9,7 +9,7 @@ authors = [
 ]
 requires-python = ">=3.11.0, <4.0.0"
 dependencies = [
-    "prefactor-core>=0.2.1",
+    "prefactor-core>=0.2.2",
     "langchain-core>=1.0.0",
 ]
 

{prefactor_langchain-0.2.2 → prefactor_langchain-0.2.3}/src/prefactor_langchain/middleware.py

@@ -526,7 +526,15 @@ class PrefactorMiddleware(AgentMiddleware):
         if loop is None or loop.is_closed():
             return ""
 
-        import asyncio
+        try:
+            running_loop = asyncio.get_running_loop()
+        except RuntimeError:
+            running_loop = None
+
+        if running_loop is not None and running_loop is loop:
+            raise RuntimeError(
+                "_create_agent_span_sync must be called from a non-event-loop thread"
+            )
 
         # Use whatever parent was set by set_parent_span() from the async side.
         # We cannot read the SpanContextStack here because run_in_executor does
@@ -664,6 +672,17 @@ class PrefactorMiddleware(AgentMiddleware):
             async def _finish() -> None:
                 await instance.finish_span(span_id, result_payload=result)
 
+            try:
+                running_loop = asyncio.get_running_loop()
+            except RuntimeError:
+                running_loop = None
+
+            if running_loop is not None and running_loop is loop:
+                raise RuntimeError(
+                    "after_agent must be called from a non-event-loop thread; "
+                    "use aafter_agent for async execution"
+                )
+
             future = asyncio.run_coroutine_threadsafe(_finish(), loop)
             future.result(timeout=5.0)
 
@@ -718,8 +737,16 @@ class PrefactorMiddleware(AgentMiddleware):
             # SpanContextStack.peek() here returns the outer workflow span
             # because we are executing in the async event loop context.
             self._agent_span_cm = instance.span("langchain:agent")
-            self._agent_span_context = await self._agent_span_cm.__aenter__()
-            await self._agent_span_context.start(params)
+            try:
+                self._agent_span_context = await self._agent_span_cm.__aenter__()
+                await self._agent_span_context.start(params)
+            except BaseException:
+                import sys
+
+                await self._agent_span_cm.__aexit__(*sys.exc_info())
+                self._agent_span_cm = None
+                self._agent_span_context = None
+                raise
 
             self._agent_span_id = self._agent_span_context.id
             logger.debug("Created agent span %s (async)", self._agent_span_id)