mirror of
https://github.com/sean1832/GPT-Brain
synced 2024-11-18 21:25:53 +00:00
24 lines
676 B
Python
24 lines
676 B
Python
|
import json
|
||
|
|
import openai
|
||
|
|
|
||
|
|
def open_file(filepath):
|
||
|
|
with open(filepath, 'r', encoding='utf-8') as file:
|
||
|
|
return file.read()
|
||
|
|
|
||
|
|
def write_file(content, filepath):
|
||
|
|
with open(filepath, 'w') as file:
|
||
|
|
file.write(content)
|
||
|
|
|
||
|
|
def write_json_file(content, filepath):
|
||
|
|
with open(filepath, 'w') as file:
|
||
|
|
json.dump(content, file, indent=2)
|
||
|
|
|
||
|
|
def read_json_file(filepath):
|
||
|
|
with open(filepath, 'r') as file:
|
||
|
|
return json.load(file)
|
||
|
|
|
||
|
|
# return a list of vectors
|
||
|
|
def embedding(content, engine='text-embedding-ada-002'):
|
||
|
|
response = openai.Embedding.create(input=content, engine=engine)
|
||
|
|
vector = response['data'][0]['embedding']
|
||
|
|
return vector
|