langchain-ai/langsmith-migrations
1
0
Fork 0
mirror of https://github.com/langchain-ai/langsmith-migrations.git synced 2026-07-01 20:03:59 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4 from langchain-ai/rahul/update-langfuse

Browse Source 
Update LangFuse functionality
This commit is contained in:
xuro-langchain
2026-04-14 11:32:59 -04:00
committed by GitHub
parent 95a246e5d7 2e9fdf74a8
commit f68c63b8c4
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
providers/langfuse/main.py
+8 -3
View File
@@ -1,4 +1,5 @@
from utils.langsmith import ls_get_or_create_workspace
from config import LS_WORKSPACE_ID
from providers.langfuse.data.prompts import migrate_prompts
from providers.langfuse.data.datasets import migrate_datasets
@@ -12,9 +13,13 @@ def migrate_langfuse(projects: list[dict]):
pid = proj.get("id") or proj.get("project_id") or proj.get("uuid") or ""
print(f"\n- Project: {pname}")
ws = ls_get_or_create_workspace(pname)
ws_id = ws["id"]
print(f" + workspace id: {ws_id}")
if LS_WORKSPACE_ID:
ws_id = LS_WORKSPACE_ID
print(f" + using workspace: {ws_id}")
else:
ws = ls_get_or_create_workspace(pname)
ws_id = ws["id"]
print(f" + workspace id: {ws_id}")
migrate_prompts(ws_id, pid)
migrate_datasets(ws_id, pid)
utils/langsmith.py
+2 -2
View File
@@ -55,7 +55,7 @@ def ls_push_prompt(
pat = pat or LS_API_KEY
session = requests.Session()
session.headers.update({"X-Tenant-Id": workspace_id})
client = Client(api_key=pat, api_url=LS_BASE, session=session)
client = Client(api_key=pat, api_url=f"{LS_BASE}/api/v1", session=session)
url = client.push_prompt(
name,
@@ -80,7 +80,7 @@ def ls_replay_runs_sdk(workspace_id: str, runs: list[dict], project_name: str =
"""
session = requests.Session()
session.headers.update({"X-Tenant-Id": workspace_id})
client = Client(api_key=LS_API_KEY, api_url=LS_BASE, session=session)
client = Client(api_key=LS_API_KEY, api_url=f"{LS_BASE}/api/v1", session=session)
def _parse_dt(val):
if not val: