CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-07-06 21:17:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

dead code

Browse Source
This commit is contained in:
Adam Wilson
2025-08-18 14:31:46 -06:00
parent fb95e8afea
commit 3d335d047c
1 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/services/nlp/text_generation_completion_service.py
+2 -8
View File
@@ -177,12 +177,6 @@ class TextGenerationCompletionService(AbstractTextGenerationCompletionService):
logger.info(f"Successfully loaded model: {model_id.value}")
def get_current_model_info(self) -> Optional[Dict[str, Any]]:
"""Get information about the currently loaded model"""
if self._current_model and self._current_model.is_loaded():
return self._current_model.get_model_info()
return None
def _process_prompt_with_guidelines_if_applicable(self, user_prompt: str, target_model_id: ModelId):
guidelines_config = (
self._use_zero_shot_chain_of_thought,
@@ -421,8 +415,8 @@ class TextGenerationCompletionService(AbstractTextGenerationCompletionService):
target_model_id = model_id or self._current_model_id or self.default_model_id
if (self._current_model_id != target_model_id or
self._current_model is None or
not self._current_model.is_loaded()):
self._current_model is None
):
self.load_model(target_model_id)
print(f'Using model: {target_model_id.value}, guidelines: {self.get_current_config()}')