Spaces:
Sleeping
Sleeping
File size: 2,995 Bytes
01d5a5d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
from typing import Optional
from lpm_kernel.api.repositories.user_llm_config_repository import UserLLMConfigRepository
from lpm_kernel.api.dto.user_llm_config_dto import (
UserLLMConfigDTO,
UpdateUserLLMConfigDTO
)
from datetime import datetime
class UserLLMConfigService:
"""User LLM Configuration Service"""
def __init__(self):
self.repository = UserLLMConfigRepository()
def get_available_llm(self) -> Optional[UserLLMConfigDTO]:
"""Get available LLM configuration
Since we only have one default configuration now (ID=1), just return it
"""
return self.repository.get_default_config()
def update_config(
self,
config_id: int,
dto: UpdateUserLLMConfigDTO
) -> UserLLMConfigDTO:
"""Update configuration or create if not exists
This method ensures that only one configuration record exists in the database.
If the configuration with the given ID doesn't exist, it will be created.
Args:
config_id: Configuration ID (should be 1)
dto: UpdateUserLLMConfigDTO object
Returns:
Updated or created configuration
"""
# Check if we need to clean up extra records
self._ensure_single_record()
# Update or create the configuration
return self.repository.update(config_id, dto)
def delete_key(self, config_id: int = 1) -> Optional[UserLLMConfigDTO]:
"""Delete API key from the configuration
This method removes the API key and related fields from the configuration.
Args:
config_id: Configuration ID (default is 1)
Returns:
Updated configuration with key removed
"""
# Check if we need to clean up extra records
self._ensure_single_record()
# Get the current configuration
config = self.repository.get_default_config()
if not config:
# If no configuration exists, return None
return None
# delete
return self.repository.delete(config_id)
def _ensure_single_record(self):
"""Ensure that only one configuration record exists in the database"""
# This is a safety measure to ensure we only have one record
# In normal operation, this should never be needed
count = self.repository.count()
if count != 1:
# If we have more than one record, we need to clean up
# This is a rare case that should not happen in normal operation
# Implementation would depend on how we want to handle this case
# For now, we'll just log a warning
from lpm_kernel.common.logging import logger
logger.warning(f"Found {count} LLM configurations in the database. Only one should exist.")
# Future implementation could delete extra records
|