mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-08-11 02:00:53 +00:00
85eb367ede
### What problem does this PR solve? #6155 Add basic Langfuse support for LLM module. A trace example: <img width="755" alt="image" src="https://github.com/user-attachments/assets/25c1f852-5116-486c-a47f-6097187142ca" /> ### Type of change - [x] New Feature (non-breaking change which adds functionality)
57 lines
2.0 KiB
Python
57 lines
2.0 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.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
|
|
def update_by_tenant(cls, tenant_id, langfuse_keys):
|
|
fields = ["tenant_id", "host", "secret_key", "public_key"]
|
|
return cls.model.update(**langfuse_keys).fields(*fields).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
|