forked from phito/darknet_diaries_llm
fix: pr comments
This commit is contained in:
parent
6d7600e47e
commit
7774fc2698
3
main.py
3
main.py
@ -17,11 +17,10 @@ load_dotenv()
|
||||
OPEN_API_KEY = os.getenv('OPEN_API_KEY')
|
||||
|
||||
# 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)
|
||||
set_global_service_context(service_context)
|
||||
|
||||
# TODO split in small functions
|
||||
if __name__ == '__main__':
|
||||
if not os.path.exists("./index/lock"):
|
||||
documents = []
|
||||
|
Loading…
Reference in New Issue
Block a user