Pārlūkot izejas kodu

Feat: add project_id and project_name to Langfuse API (#6481)

### What problem does this PR solve?

Enhance Langfuse API: add project_id and project_name

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
tags/v0.18.0
Yongteng Lei pirms 7 mēnešiem
vecāks
revīzija
542cf16292
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 14 papildinājumiem un 2 dzēšanām
  1. 14
    2
      api/apps/langfuse_app.py

+ 14
- 2
api/apps/langfuse_app.py Parādīt failu

@@ -64,7 +64,19 @@ def set_api_key():
def get_api_key():
langfuse_entry = TenantLangfuseService.filter_by_tenant_with_info(tenant_id=current_user.id)
if not langfuse_entry:
return get_error_data_result(message="Have not record any Langfuse keys.")
return get_json_result(message="Have not record any Langfuse keys.")

langfuse = Langfuse(public_key=langfuse_entry["public_key"], secret_key=langfuse_entry["secret_key"], host=langfuse_entry["host"])
try:
if not langfuse.auth_check():
return get_error_data_result(message="Invalid Langfuse keys loaded")
except langfuse.api.core.api_error.ApiError as api_err:
return get_json_result(message=f"Error from Langfuse: {api_err}")
except Exception as e:
server_error_response(e)

langfuse_entry["project_id"] = langfuse.api.projects.get().dict()["data"][0]["id"]
langfuse_entry["project_name"] = langfuse.api.projects.get().dict()["data"][0]["name"]

return get_json_result(data=langfuse_entry)

@@ -75,7 +87,7 @@ def get_api_key():
def delete_api_key():
langfuse_entry = TenantLangfuseService.filter_by_tenant(tenant_id=current_user.id)
if not langfuse_entry:
return get_error_data_result(message="Have not record any Langfuse keys.")
return get_json_result(message="Have not record any Langfuse keys.")

with DB.atomic():
try:

Notiek ielāde…
Atcelt
Saglabāt