Removed extra stuff

Author: Diophontine

src/main.py

@@ -1,42 +1 @@
 from langchain.chat_models import ChatOpenAI
-from langchain.prompts.chat import (
-    ChatPromptTemplate,
-    SystemMessagePromptTemplate,
-    HumanMessagePromptTemplate,
-)
-from langchain.chains import LLMChain
-from langchain.schema import BaseOutputParser
-
-# my first langchain
-chat_model = ChatOpenAI()
-print(chat_model.predict("hi!"))
-
-
-# create templates
-system_template = """You are a helpful assistant who generates comma separated lists.
-A user will pass in a category, and you should generated 5 objects in that category in a comma separated list.
-ONLY return a comma separated list, and nothing more."""
-system_message_prompt = SystemMessagePromptTemplate.from_template(system_template)
-
-human_template = "{text}"
-human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)
-
-# create full prompt
-chat_prompt = ChatPromptTemplate.from_messages(
-    [system_message_prompt, human_message_prompt])
-
-# process output
-class CommaSeparatedListOutputParser(BaseOutputParser):
-    """Parse the output of an LLM call to a comma-separated list."""
-
-    def parse(self, text: str):
-        """Parse the output of an LLM call."""
-        return text.strip().split(", ")
-
-# build a chain
-chain = LLMChain(
-    llm=ChatOpenAI(),
-    prompt=chat_prompt,
-    output_parser=CommaSeparatedListOutputParser()
-)
-print(chain.run("colors"))