mirror of https://github.com/sean1832/GPT-Brain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
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 |