PR feedback

Author: mfateev

langchain/langchain_interceptor.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 
+from typing import Any, Mapping, Protocol, Type
 from temporalio import workflow, worker, converter, client, api, activity
 
 with workflow.unsafe.imports_passed_through():
     from contextlib import contextmanager
-    from typing import Any, Mapping, Protocol, Type
     from langsmith import trace, tracing_context
     from langsmith.run_helpers import get_current_run_tree
 

langchain/worker.py

@@ -30,7 +30,8 @@ async def main():
 
 
 if __name__ == "__main__":
-    loop = asyncio.get_event_loop()
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
     try:
         loop.run_until_complete(main())
     except KeyboardInterrupt:

langchain/workflow.py

@@ -1,12 +1,12 @@
 
+from typing import List
+from datetime import timedelta
+import asyncio
 
 from temporalio import workflow
 
 with workflow.unsafe.imports_passed_through():
-    import asyncio
     from dataclasses import dataclass
-    from datetime import timedelta
-    from typing import List
     from activities import TranslateParams, translate_phrase
 
 @workflow.defn

pyproject.toml

@@ -38,6 +38,7 @@ gevent = ["gevent==25.4.2 ; python_version >= '3.8'"]
 langchain = [
     "langchain>=0.1.7,<0.2 ; python_version >= '3.8.1' and python_version < '4.0'",
     "langchain-openai>=0.0.6,<0.0.7 ; python_version >= '3.8.1' and python_version < '4.0'",
+    "langsmith>=0.1.22,<0.2 ; python_version >= '3.8.1' and python_version < '4.0'",
     "openai>=1.4.0,<2",
     "fastapi>=0.105.0,<0.106",
     "tqdm>=4.62.0,<5",