- RAG Specialist fully implemented new style
- Selection Specialist - VA version - fully implemented - Correction of TRAICIE_ROLE_DEFINITION_SPECIALIST - adaptation to new style - Removal of 'debug' statements
This commit is contained in:
6
common/utils/cache/translation_cache.py
vendored
6
common/utils/cache/translation_cache.py
vendored
@@ -68,7 +68,6 @@ class TranslationCacheHandler(CacheHandler[TranslationCache]):
|
||||
|
||||
setattr(translation, column.name, value)
|
||||
|
||||
current_app.logger.debug(f"Translation Cache Retrieved: {translation}")
|
||||
metrics = {
|
||||
'total_tokens': translation.prompt_tokens + translation.completion_tokens,
|
||||
'prompt_tokens': translation.prompt_tokens,
|
||||
@@ -109,7 +108,6 @@ class TranslationCacheHandler(CacheHandler[TranslationCache]):
|
||||
"""
|
||||
if not context:
|
||||
context = 'No context provided.'
|
||||
current_app.logger.debug(f"Getting translation for text: {text[:10]}..., target_lang: {target_lang}, source_lang: {source_lang}, context: {context[:10]}...")
|
||||
|
||||
def creator_func(hash_key: str) -> Optional[TranslationCache]:
|
||||
# Check if translation already exists in database
|
||||
@@ -125,8 +123,6 @@ class TranslationCacheHandler(CacheHandler[TranslationCache]):
|
||||
'time_elapsed': 0,
|
||||
'interaction_type': 'LLM'
|
||||
}
|
||||
current_app.logger.debug(f"Found existing translation in DB: {existing_translation.cache_key}")
|
||||
current_app.logger.debug(f"Metrics: {metrics}")
|
||||
current_event.log_llm_metrics(metrics)
|
||||
db.session.commit()
|
||||
return existing_translation
|
||||
@@ -165,7 +161,6 @@ class TranslationCacheHandler(CacheHandler[TranslationCache]):
|
||||
|
||||
# Generate the hash key using your existing method
|
||||
hash_key = self._generate_cache_key(text, target_lang, source_lang, context)
|
||||
current_app.logger.debug(f"Generated hash key: {hash_key}")
|
||||
|
||||
# Pass the hash_key to the get method
|
||||
return self.get(creator_func, hash_key=hash_key)
|
||||
@@ -189,7 +184,6 @@ class TranslationCacheHandler(CacheHandler[TranslationCache]):
|
||||
def translate_text(self, text_to_translate: str, target_lang: str, source_lang: str = None, context: str = None) \
|
||||
-> tuple[str, dict[str, int | float]]:
|
||||
target_language = current_app.config['SUPPORTED_LANGUAGE_ISO639_1_LOOKUP'][target_lang]
|
||||
current_app.logger.debug(f"Target language: {target_language}")
|
||||
prompt_params = {
|
||||
"text_to_translate": text_to_translate,
|
||||
"target_language": target_language,
|
||||
|
||||
Reference in New Issue
Block a user