Checker for ollama and openai added

toggle-corp · Jan 10, 2025 · dcb5697 · dcb5697
1 parent e066f2c
commit dcb5697
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/chatbotcore/contextual_chunks.py b/chatbotcore/contextual_chunks.py
@@ -92,7 +92,9 @@ def generate_contextualized_chunks(self, document: str, chunks: List[Document]):
         contextualized_chunks = []
         for chunk in chunks:
             context = self._generate_context(document, chunk.page_content)
-            context = getattr(context, "content")  # note: required when openai is used
+
+            if self.model_type == LLMType.OPENAI:
+                context = getattr(context, "content")# note: required when openai is used
 
             # Strip both context and chunk content of leading/trailing spaces
             context = context.strip()