langgraph-api 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

langgraph_api/cli.py

@@ -10,6 +10,29 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 
+def _get_org_id() -> str | None:
+    from langsmith.client import Client
+    from langsmith.utils import tracing_is_enabled
+
+    # Yes, the organizationId is actually the workspace iD
+    # which is actually the tenantID which we actually get via
+    # the sessions endpoint
+
+    if not tracing_is_enabled():
+        return
+    client = Client()
+    try:
+        response = client.request_with_retries(
+            "GET", "/api/v1/sessions", params={"limit": 1}
+        )
+        result = response.json()
+        if result:
+            return result[0]["tenant_id"]
+    except Exception as e:
+        logger.debug("Failed to get organization ID: %s", str(e))
+        return None
+
+
 @contextlib.contextmanager
 def patch_environment(**kwargs):
     """Temporarily patch environment variables.
@@ -92,14 +115,21 @@ def run_server(
     studio_url = f"https://smith.langchain.com/studio/?baseUrl={local_url}"
 
     def _open_browser():
+        nonlocal studio_url
         import time
         import urllib.request
         import webbrowser
 
+        org_id = _get_org_id()
+        if org_id:
+            studio_url = f"https://smith.langchain.com/studio/?baseUrl={local_url}&organizationId={org_id}"
+
         while True:
             try:
                 with urllib.request.urlopen(f"{local_url}/ok") as response:
                     if response.status == 200:
+                        logger.info("🎨 Opening Studio in your browser...")
+                        logger.info("URL: " + studio_url)
                         webbrowser.open(studio_url)
                         return
             except urllib.error.URLError:

langgraph_api/stream.py

@@ -94,7 +94,7 @@ async def astream_state(
 ) -> AnyStream:
     """Stream messages from the runnable."""
     run_id = str(run["run_id"])
-    await stack.enter_async_context(conn.pipeline())
+    pipe = await stack.enter_async_context(conn.pipeline())
     # extract args from run
     kwargs = run["kwargs"].copy()
     subgraphs = kwargs.get("subgraphs", False)
@@ -103,7 +103,7 @@ async def astream_state(
     graph = get_graph(
         config["configurable"]["graph_id"],
         config,
-        store=None if not conn else Store(conn),
+        store=None if not conn else Store(conn, pipe=pipe),
         checkpointer=None if temporary else Checkpointer(conn),
     )
     input = kwargs.pop("input")
@@ -188,9 +188,11 @@ async def astream_state(
                                 messages[msg.id] = msg
                                 yield "messages/metadata", {msg.id: {"metadata": meta}}
                             yield (
-                                "messages/partial"
-                                if isinstance(msg, BaseMessageChunk)
-                                else "messages/complete",
+                                (
+                                    "messages/partial"
+                                    if isinstance(msg, BaseMessageChunk)
+                                    else "messages/complete"
+                                ),
                                 [message_chunk_to_message(messages[msg.id])],
                             )
                     elif mode in stream_mode:
@@ -237,9 +239,11 @@ async def astream_state(
                             messages[msg.id] = msg
                             yield "messages/metadata", {msg.id: {"metadata": meta}}
                         yield (
-                            "messages/partial"
-                            if isinstance(msg, BaseMessageChunk)
-                            else "messages/complete",
+                            (
+                                "messages/partial"
+                                if isinstance(msg, BaseMessageChunk)
+                                else "messages/complete"
+                            ),
                             [message_chunk_to_message(messages[msg.id])],
                         )
                 elif mode in stream_mode:

{langgraph_api-0.0.3.dist-info → langgraph_api-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.3
+Version: 0.0.4
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -15,7 +15,7 @@ Requires-Dist: httpx (>=0.27.0)
 Requires-Dist: jsonschema-rs (>=0.25.0,<0.26.0)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.52,<0.3.0)
-Requires-Dist: langgraph-checkpoint (>=2.0.5,<3.0)
+Requires-Dist: langgraph-checkpoint (>=2.0.6,<3.0)
 Requires-Dist: langsmith (>=0.1.63,<0.2.0)
 Requires-Dist: orjson (>=3.10.1)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)

{langgraph_api-0.0.3.dist-info → langgraph_api-0.0.4.dist-info}/RECORD

@@ -14,7 +14,7 @@ langgraph_api/auth/langsmith/backend.py,sha256=uHeb5-h13NIjrX_LDAvfWYr3zpbJvlvbd
 langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdTV8bKd7x4Q,3998
 langgraph_api/auth/middleware.py,sha256=_gJTOskEaln4RUT2rVYdQGPJVAyAiq-orsL_eQ3CynE,1369
 langgraph_api/auth/noop.py,sha256=vDJmzG2vArJxVzdHePvrJWahEa0dvGnhc2LEMMeiFz0,391
-langgraph_api/cli.py,sha256=NNtETXM9meeGVT9nsArpuxE0Vumk1el1z3x804Esz0Q,6663
+langgraph_api/cli.py,sha256=wNGL4wnuwJn-WffuNY1akjeNSYMiZasV_uftF5XPP3E,7667
 langgraph_api/config.py,sha256=cG6eO4P_SZ2pKedb2b4n4vnBHRQr0aiECvGvOA8ZlJA,2259
 langgraph_api/cron_scheduler.py,sha256=DAzY2DsADzEpPVbG2BOSLTIufI93yeRswd71Aby_lV0,2257
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
@@ -63,7 +63,7 @@ langgraph_api/serde.py,sha256=VoJ7Z1IuqrQGXFzEP1qijAITtWCrmjtVqlCRuScjXJI,3533
 langgraph_api/server.py,sha256=cCD2lVv0SZdgf0o797UfxUyjFwmoazJVCjl_j-8Ae7A,1523
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
-langgraph_api/stream.py,sha256=N3qneAy09Wftm8Mr-7vJCA-jGXBEp2TXFi2p-QLvVzQ,11351
+langgraph_api/stream.py,sha256=yHnhODy5FRqxmiN3t0m9pBYQADws0YSHuBLet8pXBDs,11521
 langgraph_api/utils.py,sha256=FI50tOFMVidV4-1TefouL1N-OJX41qD_fSEoWigTtf0,1575
 langgraph_api/validation.py,sha256=McizHlz-Ez8Jhdbc79mbPSde7GIuf2Jlbjx2yv_l6dA,4475
 langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -75,12 +75,12 @@ langgraph_storage/database.py,sha256=0bB4y2rWBYiT4Jsmvbbsams4Q8G0CgEwPVwclz6iQrM
 langgraph_storage/ops.py,sha256=mehNrHleaPAjZ2sFISNuD7iwi_ewyX_IMmSPwreHhR0,52186
 langgraph_storage/queue.py,sha256=6cTZ0ubHu3S1T43yxHMVOwsQsDaJupByiU0sTUFFls8,3261
 langgraph_storage/retry.py,sha256=uvYFuXJ-T6S1QY1ZwkZHyZQbsvS-Ab68LSbzbUUSI2E,696
-langgraph_storage/store.py,sha256=Z033CojJb6jMZbMu3VPtwR0bFdfyfdUS8dQSaXUASYU,731
+langgraph_storage/store.py,sha256=k5DdW_dA1VaYnXo7Cm2IoXBqxm1FtPFiqNXNsmNiDUo,1452
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=JaieC_zSdQ9bzqJYdHUfCOnNt0ALBWcdj7uVjRLh9M8,122950
-langgraph_api-0.0.3.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.3.dist-info/METADATA,sha256=h9q2RmDz5Z1hCr1Tqxbd390gZzyzfUEjoFhXXW9Xz3k,3993
-langgraph_api-0.0.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.3.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.3.dist-info/RECORD,,
+langgraph_api-0.0.4.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.4.dist-info/METADATA,sha256=yNGt2rJsJVk7b-kJJEjukG8PHtBO7WCdQMK-38KoWyc,3993
+langgraph_api-0.0.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.4.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.4.dist-info/RECORD,,

langgraph_storage/store.py

@@ -1,28 +1,44 @@
 import os
+from typing import Any
 
 from langgraph.checkpoint.memory import PersistentDict
 from langgraph.store.memory import InMemoryStore
 
 
 class DiskBackedInMemStore(InMemoryStore):
-    def __init__(self, *args, filename=None, **kwargs):
+    def __init__(self, *args: Any, filename: str | None = None, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
         self.filename = filename
         self._data = PersistentDict(dict, filename=self.filename)
+        self._load_data()
+
+    def _load_data(self) -> None:
+        if not self.filename:
+            return
         try:
             self._data.load()
         except FileNotFoundError:
+            # It's okay if the file doesn't exist yet
             pass
-
-    def close(self):
+        except (EOFError, ValueError) as e:
+            raise RuntimeError(
+                f"Failed to load store from {self.filename}. "
+                "This may be due to changes in the stored data structure. "
+                "Consider clearing the local store by running: rm -rf .langgraph_api"
+            ) from e
+        except Exception as e:
+            raise RuntimeError(
+                f"Unexpected error loading store from {self.filename}: {str(e)}"
+            ) from e
+
+    def close(self) -> None:
         self._data.close()
 
 
 _STORE_FILE = os.path.join(".langgraph_api", "store.pckl")
-if not os.path.exists(".langgraph_api"):
-    os.mkdir(".langgraph_api")
+os.makedirs(".langgraph_api", exist_ok=True)
 STORE = DiskBackedInMemStore(filename=_STORE_FILE)
 
 
-def Store(*args, **kwargs):
+def Store(*args: Any, **kwargs: Any) -> DiskBackedInMemStore:
     return STORE