mirror of
https://github.com/hwchase17/langchain
synced 2024-11-20 03:25:56 +00:00
14 lines
431 B
Python
14 lines
431 B
Python
from neo4j_generation.chain import chain
|
|
|
|
if __name__ == "__main__":
|
|
text = "Harrison works at LangChain, which is located in San Francisco"
|
|
allowed_nodes = ["Person", "Organization", "Location"]
|
|
allowed_relationships = ["WORKS_AT", "LOCATED_IN"]
|
|
print( # noqa: T201
|
|
chain(
|
|
text,
|
|
allowed_nodes=allowed_nodes,
|
|
allowed_relationships=allowed_relationships,
|
|
)
|
|
)
|