File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 66from langchain .chat_models import ChatOpenAI
77from langchain .llms import OpenAI
88from langchain .chains import ConversationChain , LLMChain
9- from langchain .memory import ConversationBufferWindowMemory
109from langchain .prompts .chat import (
1110 ChatPromptTemplate ,
1211 SystemMessagePromptTemplate ,
@@ -29,7 +28,7 @@ def generate_code(human_input):
2928 system_message_prompt = SystemMessagePromptTemplate .from_template (system_prompt_template )
3029 human_message_prompt = HumanMessagePromptTemplate .from_template (human_prompt_template )
3130 # delete the gpt-4 model_name to use the default gpt-3.5 turbo for faster results
32- gpt_4 = ChatOpenAI (temperature = 0 , model_name = 'gpt-4' )
31+ gpt_4 = ChatOpenAI (temperature = 0 , model_name = 'gpt-4' ) #chat model, no memory
3332 conversation = [system_message_prompt , human_message_prompt ]
3433 chat_prompt = ChatPromptTemplate .from_messages (conversation )
3534 response = gpt_4 (chat_prompt .format_prompt (
You can’t perform that action at this time.
0 commit comments