From 2e9fdf74a81959fc64a473d70755b0adb34e6453 Mon Sep 17 00:00:00 2001 From: Rahul Verma Date: Mon, 13 Apr 2026 19:40:18 -0400 Subject: [PATCH] Update LangFuse functionality --- providers/langfuse/main.py | 11 ++++++++--- utils/langsmith.py | 4 ++-- 2 files changed, 10 insertions(+), 5 deletions(-) diff --git a/providers/langfuse/main.py b/providers/langfuse/main.py index e71b989..7389cb2 100644 --- a/providers/langfuse/main.py +++ b/providers/langfuse/main.py @@ -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) diff --git a/utils/langsmith.py b/utils/langsmith.py index d5c0d97..58d6719 100644 --- a/utils/langsmith.py +++ b/utils/langsmith.py @@ -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: