fix typos

SciPhi-AI · Oct 9, 2024 · f1678c6 · f1678c6
1 parent 0a769bc
commit f1678c6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 13 deletions.
diff --git a/docs/cookbooks/graphrag.mdx b/docs/cookbooks/graphrag.mdx
@@ -179,7 +179,7 @@ This will run a cost estimation step to give you an estimate of the cost of the
 Time taken: 0.21 seconds
 {
   "results": {
-    "message": "These are estimated ranges, actual values may vary. To run the KG creation process, run `create-graph` with `--run` in the cli, or `run_mode=\"run\"` in the client.",
+    "message": "Ran Graph Creation Estimate (not the actual run). Note that these are estimated ranges, actual values may vary. To run the KG creation process, run `create-graph` with `--run` in the cli, or `run_type=\"run\"` in the client.",
     "document_count": 2,
     "number_of_jobs_created": 3,
     "total_chunks": 29,

diff --git a/py/core/providers/kg/postgres.py b/py/core/providers/kg/postgres.py
@@ -805,7 +805,7 @@ async def get_creation_estimate(
         )  # 10 minutes per million tokens
 
         return KGCreationEstimationResponse(
-            message='These are estimated ranges, actual values may vary. To run the KG creation process, run `create-graph` with `--run` in the cli, or `run_mode="run"` in the client.',
+            message='Ran Graph Creation Estimate (not the actual run). Note that these are estimated ranges, actual values may vary. To run the KG creation process, run `create-graph` with `--run` in the cli, or `run_type="run"` in the client.',
             document_count=len(document_ids),
             number_of_jobs_created=len(document_ids) + 1,
             total_chunks=total_chunks,
@@ -879,7 +879,7 @@ async def get_enrichment_estimate(
         )
 
         return KGEnrichmentEstimationResponse(
-            message='These are estimated ranges, actual values may vary. To run the KG enrichment process, run `enrich-graph` with `--run` in the cli, or `run_mode="run"` in the client.',
+            message='Ran Graph Enrichment Estimate (not the actual run). Note that these are estimated ranges, actual values may vary. To run the KG enrichment process, run `enrich-graph` with `--run` in the cli, or `run_type="run"` in the client.',
             total_entities=entity_count,
             total_triples=triple_count,
             estimated_llm_calls=self._get_str_estimation_output(

diff --git a/py/shared/abstractions/kg.py b/py/shared/abstractions/kg.py
@@ -16,11 +16,6 @@ class KGRunType(str, Enum):
 class KGCreationSettings(R2RSerializable):
     """Settings for knowledge graph creation."""
 
-    run_mode: KGRunType = Field(
-        default=KGRunType.ESTIMATE,  # or run
-        description="Run an estimate for the full graph creation process.",
-    )
-
     kg_triples_extraction_prompt: str = Field(
         default="graphrag_triples_extraction_few_shot",
         description="The prompt to use for knowledge graph extraction.",
@@ -65,11 +60,6 @@ class KGCreationSettings(R2RSerializable):
 class KGEnrichmentSettings(R2RSerializable):
     """Settings for knowledge graph enrichment."""
 
-    run_mode: str = Field(
-        default="estimate",  # or run
-        description="Run an estimate for the full graph enrichment process.",
-    )
-
     skip_clustering: bool = Field(
         default=False,
         description="Whether to skip leiden clustering on the graph or not.",