adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(model): bug fix

Browse Source
This commit is contained in:
Timebomb2018
2026-01-29 15:25:25 +08:00
parent d291c241d5
commit ab9a97db22
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/app/services/prompt_optimizer_service.py
View File

@@ -16,7 +16,7 @@ from app.models.prompt_optimizer_model import (
PromptOptimizerSession, PromptOptimizerSession,
RoleType RoleType
) )
from app.repositories.model_repository import ModelConfigRepository from app.repositories.model_repository import ModelConfigRepository, ModelApiKeyRepository
from app.repositories.prompt_optimizer_repository import ( from app.repositories.prompt_optimizer_repository import (
PromptOptimizerSessionRepository PromptOptimizerSessionRepository
) )
@@ -168,7 +168,8 @@ class PromptOptimizerService:
logger.info(f"Prompt optimization started, user_id={user_id}, session_id={session_id}") logger.info(f"Prompt optimization started, user_id={user_id}, session_id={session_id}")
# Create LLM instance # Create LLM instance
api_config: ModelApiKey = model_config.api_keys[0] api_keys = ModelApiKeyRepository.get_by_model_config(self.db, model_config.id)
api_config: ModelApiKey = api_keys[0] if api_keys else None
llm = RedBearLLM(RedBearModelConfig( llm = RedBearLLM(RedBearModelConfig(
model_name=api_config.model_name, model_name=api_config.model_name,
provider=api_config.provider, provider=api_config.provider,