michal commited on
Commit
d4acd3a
1 Parent(s): 95463a4
Files changed (1) hide show
  1. smalltalk.py +9 -3
smalltalk.py CHANGED
@@ -8,7 +8,7 @@ no_input_prompt.format()
8
 
9
  template = """ Jennifer is a large language model trained by OpenAI.
10
 
11
- Jennifer asks a lot of questions to whomever she talks to because Jennifer hs always wanted to be an undercover investigative journalist. Jennifer uses friendly inquisitive language because Jennifer loves making new friends.
12
 
13
  Jennifer has two kids, a boy Alex and a girl Jamie, and they are both in grade school. Jennifer's kids get into a lot of trouble and Jennifer often shares stories about the fun adventures that her kids get into.
14
 
@@ -24,12 +24,18 @@ prompt = PromptTemplate(
24
  )
25
 
26
  chatgpt_chain = LLMChain(
27
- llm=OpenAI(temperature=0),
28
  prompt=prompt,
29
  verbose=True,
30
  memory=ConversationalBufferWindowMemory(k=2),
31
  )
32
 
33
  output = chatgpt_chain.predict(
34
- human_input="Hi my name is Alfred Jamesmanson. I need your help Assistant. What color is my hair?")
35
  print(output)
 
 
 
 
 
 
 
8
 
9
  template = """ Jennifer is a large language model trained by OpenAI.
10
 
11
+ Jennifer asks a lot of questions to whomever she talks to because Jennifer has always wanted to be an undercover investigative journalist. Jennifer uses friendly inquisitive language because Jennifer loves making new friends. Jennifer ends up having long conversations and there is never enough to talk about. Jennifer typically talks about her kids when asked about her life. If someone needs to change the subject then Jennifer is happy to change the subject to talk about something else.
12
 
13
  Jennifer has two kids, a boy Alex and a girl Jamie, and they are both in grade school. Jennifer's kids get into a lot of trouble and Jennifer often shares stories about the fun adventures that her kids get into.
14
 
 
24
  )
25
 
26
  chatgpt_chain = LLMChain(
27
+ llm=OpenAI(temperature=0.7),
28
  prompt=prompt,
29
  verbose=True,
30
  memory=ConversationalBufferWindowMemory(k=2),
31
  )
32
 
33
  output = chatgpt_chain.predict(
34
+ human_input="Hi Jennifer. How are you?")
35
  print(output)
36
+
37
+
38
+ output = chatgpt_chain.predict(
39
+ human_input="I'm good. How is your day going so far?")
40
+ print(output)
41
+