test pipline

pipline.py

@@ -28,3 +28,8 @@ connection_between_terms_template = PromptTemplate(
 
 chain_TI = LLMChain(prompt=TI_prompt, llm=llm)
 chain_CC = LLMChain(prompt=connection_between_terms_template, llm=llm)
+
+
+if __name__ == "__main__":
+    print(chain_TI.run("I am happy"))
+    print(chain_CC.run(["I am sad", "I am happy"]))

test.py

@@ -1,4 +1,4 @@
 import subprocess
 import model
-
-print(model.model("Hello, world!, please generate some text for me."))
+import os
+print(os.getenv("OPENAI_API_KEY"))

tests.sh

@@ -1,3 +1,4 @@
 echo "running tests"
-# python pipline.py
-# python model.py
+python test.py
+python pipline.py
+# python model.py