mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-05-04 07:05:27 +08:00
542cf16292
### 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)
98 lines
3.7 KiB
Python
98 lines
3.7 KiB
Python
#
|
|
# Copyright 2025 The InfiniFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
|
|
from flask import request
|
|
from flask_login import current_user, login_required
|
|
from langfuse import Langfuse
|
|
|
|
from api.db.db_models import DB
|
|
from api.db.services.langfuse_service import TenantLangfuseService
|
|
from api.utils.api_utils import get_error_data_result, get_json_result, server_error_response, validate_request
|
|
|
|
|
|
@manager.route("/api_key", methods=["POST", "PUT"]) # noqa: F821
|
|
@login_required
|
|
@validate_request("secret_key", "public_key", "host")
|
|
def set_api_key():
|
|
req = request.get_json()
|
|
secret_key = req.get("secret_key", "")
|
|
public_key = req.get("public_key", "")
|
|
host = req.get("host", "")
|
|
if not all([secret_key, public_key, host]):
|
|
return get_error_data_result(message="Missing required fields")
|
|
|
|
langfuse_keys = dict(
|
|
tenant_id=current_user.id,
|
|
secret_key=secret_key,
|
|
public_key=public_key,
|
|
host=host,
|
|
)
|
|
|
|
langfuse = Langfuse(public_key=langfuse_keys["public_key"], secret_key=langfuse_keys["secret_key"], host=langfuse_keys["host"])
|
|
if not langfuse.auth_check():
|
|
return get_error_data_result(message="Invalid Langfuse keys")
|
|
|
|
langfuse_entry = TenantLangfuseService.filter_by_tenant(tenant_id=current_user.id)
|
|
with DB.atomic():
|
|
try:
|
|
if not langfuse_entry:
|
|
TenantLangfuseService.save(**langfuse_keys)
|
|
else:
|
|
TenantLangfuseService.update_by_tenant(tenant_id=current_user.id, langfuse_keys=langfuse_keys)
|
|
return get_json_result(data=langfuse_keys)
|
|
except Exception as e:
|
|
server_error_response(e)
|
|
|
|
|
|
@manager.route("/api_key", methods=["GET"]) # noqa: F821
|
|
@login_required
|
|
@validate_request()
|
|
def get_api_key():
|
|
langfuse_entry = TenantLangfuseService.filter_by_tenant_with_info(tenant_id=current_user.id)
|
|
if not langfuse_entry:
|
|
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)
|
|
|
|
|
|
@manager.route("/api_key", methods=["DELETE"]) # noqa: F821
|
|
@login_required
|
|
@validate_request()
|
|
def delete_api_key():
|
|
langfuse_entry = TenantLangfuseService.filter_by_tenant(tenant_id=current_user.id)
|
|
if not langfuse_entry:
|
|
return get_json_result(message="Have not record any Langfuse keys.")
|
|
|
|
with DB.atomic():
|
|
try:
|
|
TenantLangfuseService.delete_model(langfuse_entry)
|
|
return get_json_result(data=True)
|
|
except Exception as e:
|
|
server_error_response(e)
|