fix black

ansible · Oct 17, 2024 · 77fb6a2 · 77fb6a2
1 parent 7d0a210
commit 77fb6a2
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 11 deletions.
diff --git a/ols/app/models/config.py b/ols/app/models/config.py
@@ -902,9 +902,11 @@ def __init__(
         if data is None:
             return
 
-        self.conversation_cache = ConversationCacheConfig(
-            data.get("conversation_cache")
-        ) if data.get("conversation_cache") else None
+        self.conversation_cache = (
+            ConversationCacheConfig(data.get("conversation_cache"))
+            if data.get("conversation_cache")
+            else None
+        )
         self.logging_config = LoggingConfig(**data.get("logging_config", {}))
         if data.get("reference_content") is not None:
             self.reference_content = ReferenceContent(data.get("reference_content"))
@@ -933,7 +935,7 @@ def __init__(
         self.certificate_directory = data.get(
             "certificate_directory", constants.DEFAULT_CERTIFICATE_DIRECTORY
         )
-        self.customize = data.get('customize')
+        self.customize = data.get("customize")
 
     def __eq__(self, other: object) -> bool:
         """Compare two objects for equality."""

diff --git a/ols/customize/__init__.py b/ols/customize/__init__.py
@@ -1,6 +1,6 @@
 import os
 import importlib
 
-project = os.getenv('PROJECT', 'ols')
-prompts = importlib.import_module(f'ols.customize.{project}.prompts')
-keywords = importlib.import_module(f'ols.customize.{project}.keywords')
+project = os.getenv("PROJECT", "ols")
+prompts = importlib.import_module(f"ols.customize.{project}.prompts")
+keywords = importlib.import_module(f"ols.customize.{project}.keywords")
diff --git a/ols/src/prompts/prompt_generator.py b/ols/src/prompts/prompt_generator.py
@@ -12,6 +12,7 @@
 from ols.constants import ModelFamily
 from ols.customize import prompts
 
+
 def restructure_rag_context_pre(text: str, model: str) -> str:
     """Restructure rag text - pre truncation."""
     if ModelFamily.GRANITE in model:
@@ -62,7 +63,9 @@ def _generate_prompt_gpt(self) -> tuple[ChatPromptTemplate, dict]:
 
         if len(self._rag_context) > 0:
             llm_input_values["context"] = "".join(self._rag_context)
-            sys_intruction = sys_intruction + "\n" + prompts.USE_CONTEXT_INSTRUCTION.strip()
+            sys_intruction = (
+                sys_intruction + "\n" + prompts.USE_CONTEXT_INSTRUCTION.strip()
+            )
 
         if len(self._history) > 0:
             chat_history = []
@@ -73,7 +76,9 @@ def _generate_prompt_gpt(self) -> tuple[ChatPromptTemplate, dict]:
                     chat_history.append(AIMessage(content=h.removeprefix("ai: ")))
             llm_input_values["chat_history"] = chat_history
 
-            sys_intruction = sys_intruction + "\n" + prompts.USE_HISTORY_INSTRUCTION.strip()
+            sys_intruction = (
+                sys_intruction + "\n" + prompts.USE_HISTORY_INSTRUCTION.strip()
+            )
 
         if "context" in llm_input_values:
             sys_intruction = sys_intruction + "\n{context}"
@@ -93,10 +98,14 @@ def _generate_prompt_granite(self) -> tuple[PromptTemplate, dict]:
 
         if len(self._rag_context) > 0:
             llm_input_values["context"] = "".join(self._rag_context)
-            prompt_message = prompt_message + "\n" + prompts.USE_CONTEXT_INSTRUCTION.strip()
+            prompt_message = (
+                prompt_message + "\n" + prompts.USE_CONTEXT_INSTRUCTION.strip()
+            )
 
         if len(self._history) > 0:
-            prompt_message = prompt_message + "\n" + prompts.USE_HISTORY_INSTRUCTION.strip()
+            prompt_message = (
+                prompt_message + "\n" + prompts.USE_HISTORY_INSTRUCTION.strip()
+            )
             llm_input_values["chat_history"] = "".join(self._history)
 
         if "context" in llm_input_values: