smalltalk

smalltalk.py

@@ -0,0 +1,35 @@
+
+from langchain import OpenAI, ConversationChain, LLMChain, PromptTemplate
+from langchain.chains.conversation.memory import ConversationalBufferWindowMemory
+from langchain import PromptTemplate
+
+no_input_prompt = PromptTemplate(input_variables=[], template="Tell me a joke.")
+no_input_prompt.format()
+
+template = """ Jennifer is a large language model trained by OpenAI.
+
+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.
+
+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.
+
+Jennifer is able to process and understand large amounts of text, and can use this knowledge to provide accurate and informative responses to a wide range of questions. Additionally, Jennifer  is able to generate its own text based on the input it receives, allowing it to engage in discussions and provide explanations and descriptions on a wide range of topics.
+
+human: {human_input}
+Jennifer:
+"""
+
+prompt = PromptTemplate(
+    input_variables=["human_input"], 
+    template=template
+)
+
+chatgpt_chain = LLMChain(
+    llm=OpenAI(temperature=0), 
+    prompt=prompt, 
+    verbose=True, 
+    memory=ConversationalBufferWindowMemory(k=2),
+)
+
+output = chatgpt_chain.predict(
+  human_input="Hi my name is Alfred Jamesmanson. I need your help Assistant. What color is my hair?")
+print(output)