mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
14 lines
417 B
Python
14 lines
417 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(
|
||
|
|
chain(
|
||
|
|
text,
|
||
|
|
allowed_nodes=allowed_nodes,
|
||
|
|
allowed_relationships=allowed_relationships,
|
||
|
|
)
|
||
|
|
)
|