Skip to content

Commit

Permalink
Remove datashaper strip code (#1581)
Browse files Browse the repository at this point in the history
Remove datashaper
  • Loading branch information
natoverse authored Jan 3, 2025
1 parent 58f646a commit a35cb12
Show file tree
Hide file tree
Showing 151 changed files with 2,033 additions and 4,066 deletions.
4 changes: 4 additions & 0 deletions .semversioner/next-release/minor-20241231213627966329.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
{
"type": "minor",
"description": "Remove old pipeline runner."
}
4 changes: 4 additions & 0 deletions .semversioner/next-release/minor-20241231214323349946.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
{
"type": "minor",
"description": "Remove DataShaper (first steps)."
}
4 changes: 0 additions & 4 deletions dictionary.txt
Original file line number Diff line number Diff line change
Expand Up @@ -148,10 +148,6 @@ codebases
# Microsoft
MSRC

# Broken Upstream
# TODO FIX IN DATASHAPER
Arrary

# Prompt Inputs
ABILA
Abila
Expand Down
3 changes: 1 addition & 2 deletions docs/examples_notebooks/index_migration.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -206,9 +206,8 @@
"metadata": {},
"outputs": [],
"source": [
"from datashaper import NoopVerbCallbacks\n",
"\n",
"from graphrag.cache.factory import create_cache\n",
"from graphrag.callbacks.noop_verb_callbacks import NoopVerbCallbacks\n",
"from graphrag.index.flows.generate_text_embeddings import generate_text_embeddings\n",
"\n",
"# We only need to re-run the embeddings workflow, to ensure that embeddings for all required search fields are in place\n",
Expand Down
28 changes: 2 additions & 26 deletions docs/index/architecture.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,33 +8,9 @@ In order to support the GraphRAG system, the outputs of the indexing engine (in
This model is designed to be an abstraction over the underlying data storage technology, and to provide a common interface for the GraphRAG system to interact with.
In normal use-cases the outputs of the GraphRAG Indexer would be loaded into a database system, and the GraphRAG's Query Engine would interact with the database using the knowledge model data-store types.

### DataShaper Workflows

GraphRAG's Indexing Pipeline is built on top of our open-source library, [DataShaper](https://github.com/microsoft/datashaper).
DataShaper is a data processing library that allows users to declaratively express data pipelines, schemas, and related assets using well-defined schemas.
DataShaper has implementations in JavaScript and Python, and is designed to be extensible to other languages.

One of the core resource types within DataShaper is a [Workflow](https://github.com/microsoft/datashaper/blob/main/javascript/schema/src/workflow/WorkflowSchema.ts).
Workflows are expressed as sequences of steps, which we call [verbs](https://github.com/microsoft/datashaper/blob/main/javascript/schema/src/workflow/verbs.ts).
Each step has a verb name and a configuration object.
In DataShaper, these verbs model relational concepts such as SELECT, DROP, JOIN, etc.. Each verb transforms an input data table, and that table is passed down the pipeline.

```mermaid
---
title: Sample Workflow
---
flowchart LR
input[Input Table] --> select[SELECT] --> join[JOIN] --> binarize[BINARIZE] --> output[Output Table]
```

### LLM-based Workflow Steps

GraphRAG's Indexing Pipeline implements a handful of custom verbs on top of the standard, relational verbs that our DataShaper library provides. These verbs give us the ability to augment text documents with rich, structured data using the power of LLMs such as GPT-4. We utilize these verbs in our standard workflow to extract entities, relationships, claims, community structures, and community reports and summaries. This behavior is customizable and can be extended to support many kinds of AI-based data enrichment and extraction tasks.

### Workflow Graphs
### Workflows

Because of the complexity of our data indexing tasks, we needed to be able to express our data pipeline as series of multiple, interdependent workflows.
In the GraphRAG Indexing Pipeline, each workflow may define dependencies on other workflows, effectively forming a directed acyclic graph (DAG) of workflows, which is then used to schedule processing.

```mermaid
---
Expand All @@ -55,7 +31,7 @@ stateDiagram-v2
The primary unit of communication between workflows, and between workflow steps is an instance of `pandas.DataFrame`.
Although side-effects are possible, our goal is to be _data-centric_ and _table-centric_ in our approach to data processing.
This allows us to easily reason about our data, and to leverage the power of dataframe-based ecosystems.
Our underlying dataframe technology may change over time, but our primary goal is to support the DataShaper workflow schema while retaining single-machine ease of use and developer ergonomics.
Our underlying dataframe technology may change over time, but our primary goal is to support the workflow schema while retaining single-machine ease of use and developer ergonomics.

### LLM Caching

Expand Down
19 changes: 0 additions & 19 deletions examples/README.md

This file was deleted.

2 changes: 0 additions & 2 deletions examples/__init__.py

This file was deleted.

2 changes: 0 additions & 2 deletions examples/custom_input/__init__.py

This file was deleted.

24 changes: 0 additions & 24 deletions examples/custom_input/pipeline.yml

This file was deleted.

46 changes: 0 additions & 46 deletions examples/custom_input/run.py

This file was deleted.

2 changes: 0 additions & 2 deletions examples/single_verb/__init__.py

This file was deleted.

3 changes: 0 additions & 3 deletions examples/single_verb/input/data.csv

This file was deleted.

12 changes: 0 additions & 12 deletions examples/single_verb/pipeline.yml

This file was deleted.

77 changes: 0 additions & 77 deletions examples/single_verb/run.py

This file was deleted.

2 changes: 0 additions & 2 deletions examples/use_built_in_workflows/__init__.py

This file was deleted.

23 changes: 0 additions & 23 deletions examples/use_built_in_workflows/pipeline.yml

This file was deleted.

Loading

0 comments on commit a35cb12

Please sign in to comment.