diff --git a/src/vmk_data_collector/services/ai_enricher.py b/src/vmk_data_collector/services/ai_enricher.py index 0fe73cc..7bdd57c 100644 --- a/src/vmk_data_collector/services/ai_enricher.py +++ b/src/vmk_data_collector/services/ai_enricher.py @@ -26,7 +26,6 @@ "image_analysis_results": {"общее_впечатление": "хорошее"}, "generated_description": "Краткое привлекательное описание для покупателя...", "summary": "Краткая сводка: что за объект, цена, состояние, плюсы/минусы.", - "model_version": "llama3.2", "processing_time_ms": 1200 } Оценка качества объявления (listing_quality_score): 1–10. @@ -48,7 +47,6 @@ "image_analysis_results": {}, "generated_description": "Уютная двухкомнатная квартира в центре города.", "summary": "Квартира 50 м², 2 комнаты, цена адекватна.", - "model_version": "llama3.2-mock", "processing_time_ms": 0, } @@ -64,7 +62,9 @@ ) -> AiEnrichmentResult: if settings.ollama_mock: logger.info("ai_enricher_mock_mode") - return AiEnrichmentResult(**_MOCK_RESPONSE) + mock = dict(_MOCK_RESPONSE) + mock["model_version"] = settings.ollama_text_model + return AiEnrichmentResult(**mock) text = self._build_prompt(normalized, image_analysis_results) messages = [ @@ -85,6 +85,7 @@ ) content = response["message"]["content"] data = json.loads(content) + data["model_version"] = settings.ollama_text_model return AiEnrichmentResult(**data) except OllamaRetryableError: raise