-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.py
18 lines (14 loc) · 1 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
openai_api_key = "replace_with_your_API_Key"
neo4j_uri = "replace_with_your_neo4j_uri"
neo4j_username = "replace_with_your_neo4j_username"
neo4j_password = "replace_with_your_neo4j_password"
prompt = "***Task***"
prompt += "For the text below extract materials, compositions, properties, applications, methods and create a JSON object in this format."
prompt += "Relationships or Edges can be contains, has_material and has_property, has_application, has_value."
prompt += "Node labels can be contains, material, property, applications, compositions."
prompt += "Nodes should be with ids, names, labels, attributes and relationships should be source, target, type. Ids need to be numerical values. Json keys need to be nodes and edges."
prompt += "***Format***"
prompt += "'nodes': [{'id': '','name': '','label':'material','attributes': {}],'edges': [{'source': '','target': '','type': ''}],}"
prompt += "***Text***"
system_role = "You extract information from documents and return json objects"
model = "gpt-3.5-turbo"