-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathloader.py
More file actions
29 lines (24 loc) · 878 Bytes
/
loader.py
File metadata and controls
29 lines (24 loc) · 878 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import os
import logging
from slack_bolt import App
from dotenv import load_dotenv
from langchain import OpenAI, LLMChain, PromptTemplate
from langchain.chains.conversation.memory import ConversationBufferWindowMemory
load_dotenv()
app = App(token=os.getenv('slack_bot_token'))
logging.basicConfig(filename=os.getenv('logfile_path'), filemode='a', datefmt='%H:%M:%S',
format='%(asctime)s\t[%(name)s]\t%(levelname)s\t%(message)s', level=logging.WARNING)
template = open(os.getenv('template_path')).read()
prompt = PromptTemplate(
input_variables=['history', 'human_input'],
template=template
)
gpt_chain = LLMChain(
llm=OpenAI(
temperature=0,
model=os.getenv('ai_model'),
),
prompt=prompt,
memory=ConversationBufferWindowMemory(k=int(os.getenv('buffer_message_count'))),
verbose=False, # Logging to console
)