mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-04-23 14:40:03 +08:00
5e0a77df2b
### What problem does this PR solve? Add Langfuse APIs ### Type of change - [x] New Feature (non-breaking change which adds functionality)
72 lines
2.6 KiB
Python
72 lines
2.6 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 datetime import datetime
|
|
|
|
import peewee
|
|
|
|
from api.db.db_models import DB, TenantLangfuse
|
|
from api.db.services.common_service import CommonService
|
|
from api.utils import current_timestamp, datetime_format
|
|
|
|
|
|
class TenantLangfuseService(CommonService):
|
|
"""
|
|
All methods that modify the status should be enclosed within a DB.atomic() context to ensure atomicity
|
|
and maintain data integrity in case of errors during execution.
|
|
"""
|
|
|
|
model = TenantLangfuse
|
|
|
|
@classmethod
|
|
@DB.connection_context()
|
|
def filter_by_tenant(cls, tenant_id):
|
|
fields = [cls.model.tenant_id, cls.model.host, cls.model.secret_key, cls.model.public_key]
|
|
try:
|
|
keys = cls.model.select(*fields).where(cls.model.tenant_id == tenant_id).first()
|
|
return keys
|
|
except peewee.DoesNotExist:
|
|
return None
|
|
|
|
@classmethod
|
|
@DB.connection_context()
|
|
def filter_by_tenant_with_info(cls, tenant_id):
|
|
fields = [cls.model.tenant_id, cls.model.host, cls.model.secret_key, cls.model.public_key]
|
|
try:
|
|
keys = cls.model.select(*fields).where(cls.model.tenant_id == tenant_id).dicts().first()
|
|
return keys
|
|
except peewee.DoesNotExist:
|
|
return None
|
|
|
|
@classmethod
|
|
def update_by_tenant(cls, tenant_id, langfuse_keys):
|
|
langfuse_keys["update_time"] = current_timestamp()
|
|
langfuse_keys["update_date"] = datetime_format(datetime.now())
|
|
return cls.model.update(**langfuse_keys).where(cls.model.tenant_id == tenant_id).execute()
|
|
|
|
@classmethod
|
|
def save(cls, **kwargs):
|
|
kwargs["create_time"] = current_timestamp()
|
|
kwargs["create_date"] = datetime_format(datetime.now())
|
|
kwargs["update_time"] = current_timestamp()
|
|
kwargs["update_date"] = datetime_format(datetime.now())
|
|
obj = cls.model.create(**kwargs)
|
|
return obj
|
|
|
|
@classmethod
|
|
def delete_model(cls, langfuse_model):
|
|
langfuse_model.delete_instance()
|