Metadata-Version: 2.4
Name: kweaver-data-retrieval
Version: 0.1.2
Summary: A library for Kweaver data retrieval
Author-email: Chen Xiao <xavier.chen@aishu.cn>
License: Apache-2.0
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.10
Description-Content-Type: text/markdown
Requires-Dist: Jinja2==3.1.3
Requires-Dist: langchain==0.2.1
Requires-Dist: langchain_community==0.2.1
Requires-Dist: langchain_core==0.2.3
Requires-Dist: langchain_openai==0.1.8
Requires-Dist: Requests==2.32.3
Requires-Dist: aiohttp==3.9.4
Requires-Dist: pydantic==2.7.3
Requires-Dist: pandas==2.2.2
Requires-Dist: async-timeout==4.0.3
Requires-Dist: pycryptodome==3.20.0
Requires-Dist: urllib3==2.1.0
Requires-Dist: altair==5.4.0
Requires-Dist: commentjson==0.9.0
Requires-Dist: plotly==5.23.0
Requires-Dist: jsonref==1.1.0
Requires-Dist: vegafusion==1.6.9
Requires-Dist: tabulate==0.9.0
Requires-Dist: redis==5.0.8
Requires-Dist: pydantic-settings==2.1.0
Requires-Dist: sqlparse==0.5.1
Requires-Dist: sql_metadata==2.13.0
Requires-Dist: pyahocorasick==2.0.0
Requires-Dist: jieba==0.42.1
Requires-Dist: faiss-cpu>=1.8
Requires-Dist: fastapi>=0.115.0
Requires-Dist: openai>=1.68.2
Requires-Dist: httpx==0.27.2
Requires-Dist: jupyter-kernel-gateway==3.0.1
Requires-Dist: sqlglot[rs]==26.21.0
Requires-Dist: nebula3-python==3.5.0
Requires-Dist: rank-bm25==0.2.2
Provides-Extra: linux
Requires-Dist: gunicorn>=21.2.0; extra == "linux"
Provides-Extra: windows

# kweaver-data-retrieval

A library for Kweaver data retrieval.

## Installation

```bash
pip install kweaver-data-retrieval
```

## Usage

### Import Tools

```python
from data_retrieval.tools import TOOLS_MAPPING

# List available tools
print(TOOLS_MAPPING.keys())
```

### Knowledge Network Tools

```python
from data_retrieval.tools.knowledge_network_tools import KNOWLEDGE_NETWORK_TOOLS_MAPPING

# Available tools:
# - knowledge_rerank: Rerank search results
# - knowledge_retrieve: Retrieve from knowledge base
# - kn_search: Knowledge network search (v2)
# - kn_path_search: Relation path retrieval
# - cypher_query: Execute Cypher queries
```

### Start API Server

```python
from data_retrieval.tools.tool_api_router import DEFAULT_APP
import uvicorn

uvicorn.run(DEFAULT_APP, host="0.0.0.0", port=9100)
```

Or from command line:

```bash
uvicorn data_retrieval.tools.tool_api_router:DEFAULT_APP --host 0.0.0.0 --port 9100
```

## Documentation

- [GitHub Repository](https://github.com/kweaver-ai/decision-agent/tree/main/data-retrieval)
- [API Documentation](https://github.com/kweaver-ai/decision-agent/blob/main/data-retrieval/api.json)

## License

Apache License 2.0
