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')
|
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 = []
|
||||||
|
Loading…
Reference in New Issue
Block a user