more testing

SciPhi-AI · Oct 20, 2024 · 03e61bf · 03e61bf
1 parent ecc3108
commit 03e61bf
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/py/core/pipes/kg/deduplication_summary.py b/py/core/pipes/kg/deduplication_summary.py
@@ -59,20 +59,20 @@ async def _merge_entity_descriptions_llm_prompt(
                 > self.kg_provider.config.kg_entity_deduplication_settings.max_description_input_length
             ):
                 break
-            index += 1
             description_length += len(entity_descriptions[index])
+            index += 1
 
         completion = await self.llm_provider.aget_completion(
             messages=await self.prompt_provider._get_message_payload(
                 task_prompt_name=self.kg_provider.config.kg_entity_deduplication_settings.kg_entity_deduplication_prompt,
                 task_inputs={
-                    "entity_name": (entity_name,),
-                    "entity_descriptions": (
-                        "\n".join(entity_descriptions[:index]),
+                    "entity_name": entity_name,
+                    "entity_descriptions": "\n".join(
+                        entity_descriptions[:index]
                     ),
                 },
             ),
-            generation_config=generation_config,
+            generation_config=GenerationConfig(**generation_config),  # type: ignore
         )
 
         # get the $$description$$