Skip to content

Commit

Permalink
fix bugs
Browse files Browse the repository at this point in the history
  • Loading branch information
emrgnt-cmplxty committed Dec 5, 2024
1 parent 6e258a1 commit 2c20278
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions py/core/main/orchestration/hatchet/kg_workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -288,19 +288,14 @@ async def kg_extraction(self, context: Context) -> dict:
input_data = get_input_data_dict(request)
document_id = input_data.get("document_id", None)
collection_id = input_data.get("collection_id", None)
# if collection_id and document_id:
# raise R2RException(
# "Both collection_id and document_id were provided. Please provide only one.",
# 400,
# )

if collection_id and not document_id:
document_ids = (
await self.kg_service.get_document_ids_for_create_graph(
collection_id=collection_id,
**input_data["graph_creation_settings"],
)
)
print("document ids to extract = ", document_ids)
workflows = []

for document_id in document_ids:
Expand Down

0 comments on commit 2c20278

Please sign in to comment.