phito/darknet_diaries_llm
2
0
Fork 1
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

fix: pr comments

Browse Source
This commit is contained in:
NIHART, Jeremi 2023-10-09 10:06:56 +02:00
parent 6d7600e47e
commit 7774fc2698
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@ -17,11 +17,10 @@ load_dotenv()
OPEN_API_KEY = os.getenv('OPEN_API_KEY') OPEN_API_KEY = os.getenv('OPEN_API_KEY')
# config llm context # config llm context
llm = OpenAI(model="gpt-4", temperature=0, max_tokens=256, api_key="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") llm = OpenAI(model="gpt-4", temperature=0, max_tokens=256, api_key=OPEN_API_KEY)
service_context = ServiceContext.from_defaults(llm=llm) service_context = ServiceContext.from_defaults(llm=llm)
set_global_service_context(service_context) set_global_service_context(service_context)
# TODO split in small functions
if __name__ == '__main__': if __name__ == '__main__':
if not os.path.exists("./index/lock"): if not os.path.exists("./index/lock"):
documents = [] documents = []