Skip to content

Commit

Permalink
Add dev instructions to README
Browse files Browse the repository at this point in the history
  • Loading branch information
oskarhane authored and willtai committed Apr 19, 2024
1 parent 5cd31c1 commit c93ff58
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 2 deletions.
24 changes: 22 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,9 @@ pip install neo4j-genai
```

## Example

After setting up a Neo4j database instance:

```python
from neo4j import GraphDatabase
from neo4j_genai import VectorRetriever
Expand Down Expand Up @@ -64,6 +66,24 @@ print(retriever.search(query_vector=query_vector, top_k=5))

```

# Development

## Install dependencies

```bash
poetry install
```

## Run tests

Open a new virtual enviroment and then run the tests.

```bash
poetry shell
pytest
```

## Further information
- [The official Neo4j Python driver](https://github.com/neo4j/neo4j-python-driver)
- [Neo4j GenAI integrations](https://neo4j.com/docs/cypher-manual/current/genai-integrations/)

- [The official Neo4j Python driver](https://github.com/neo4j/neo4j-python-driver)
- [Neo4j GenAI integrations](https://neo4j.com/docs/cypher-manual/current/genai-integrations/)
1 change: 1 addition & 0 deletions src/neo4j_genai/retrievers.py
Original file line number Diff line number Diff line change
Expand Up @@ -160,6 +160,7 @@ def search(
query_vector (Optional[list[float]], optional): The vector embeddings to get the closest neighbors of. Defaults to None.
query_text (Optional[str], optional): The text to get the closest neighbors of. Defaults to None.
top_k (int, optional): The number of neighbors to return. Defaults to 5.
query_params (Optional[dict[str, Any]], optional): Parameters for the Cypher query. Defaults to None.
Raises:
ValueError: If validation of the input arguments fail.
Expand Down

0 comments on commit c93ff58

Please sign in to comment.