update env vars for hosted langserve (#347)

Modifies the env vars we send to hosted langserve to uniquely identify the piece of code that created the env var
langchain-ai · Dec 24, 2023 · 76669c2 · 76669c2
1 parent cf180b8
commit 76669c2
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/langserve/api_handler.py b/langserve/api_handler.py
@@ -173,15 +173,18 @@ def _update_config_with_defaults(
 
     if _is_hosted():
         hosted_metadata = {
-            "__langserve_hosted_git_commit_sha": os.environ.get(
-                "HOSTED_LANGSERVE_GIT_COMMIT", ""
+            "__langserve_hosted_git_ref": os.environ.get(
+                "HOSTED_LANGSERVE_GIT_REF", ""
             ),
             "__langserve_hosted_repo_subdirectory_path": os.environ.get(
                 "HOSTED_LANGSERVE_GIT_REPO_PATH", ""
             ),
             "__langserve_hosted_repo_url": os.environ.get(
                 "HOSTED_LANGSERVE_GIT_REPO", ""
             ),
+            "__langserve_hosted_revision_id": os.environ.get(
+                "HOSTED_LANGSERVE_REVISION_ID", ""
+            ),
             "__langserve_hosted_is_hosted": "true",
         }
         metadata.update(hosted_metadata)