rm duplicates

SciPhi-AI · Oct 4, 2024 · 772a77e · 772a77e
1 parent 67ca1a3
commit 772a77e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/py/core/main/orchestration/hatchet/kg_workflow.py b/py/core/main/orchestration/hatchet/kg_workflow.py
@@ -41,10 +41,6 @@ class KGExtractDescribeEmbedWorkflow:
         def __init__(self, kg_service: KgService):
             self.kg_service = kg_service
 
-        @orchestration_provider.concurrency(max_runs=10, limit_strategy=ConcurrencyLimitStrategy.GROUP_ROUND_ROBIN)
-        def concurrency(self, context) -> str:
-            return str(context.workflow_input()["request"]["collection_id"])
-
         @orchestration_provider.concurrency(
             max_runs=orchestration_provider.config.kg_creation_concurrency_limit,
             limit_strategy=ConcurrencyLimitStrategy.GROUP_ROUND_ROBIN,

diff --git a/py/r2r.toml b/py/r2r.toml
@@ -66,7 +66,7 @@ batch_size = 256
     fragment_merge_count = 4 # number of fragments to merge into a single extraction
     max_knowledge_triples = 100
     max_description_input_length = 1024
-    generation_config = { model = "azure/gpt-4o-mini" } # and other params, model used for triplet extraction
+    generation_config = { model = "openai/gpt-4o-mini" } # and other params, model used for triplet extraction
 
   [kg.kg_enrichment_settings]
     community_reports_prompt = "graphrag_community_reports_prompt"