Update app.py

app.py

@@ -17,12 +17,13 @@ nest_asyncio.apply()
 
 # OpenAI credentials
 # key = os.getenv('OPENAI_API_KEY')
-key = os.getenv('CLAUDE_API_KEY')
-print(key)
-os.environ["ANTHROPIC_API_KEY"] = key
 # openai.api_key = key 
 # os.environ["OPENAI_API_KEY"] = key
 
+# Anthropic credentials
+key = os.getenv('CLAUDE_API_KEY')
+os.environ["ANTHROPIC_API_KEY"] = key
+
 # Streamlit UI
 st.title("Auto Test Case Generation using LLM")
 
@@ -69,6 +70,7 @@ if uploaded_files:
             # return response.choices[0].message.content
 
             response = [
+                ChatMessage(role="system", content="You are a sincere and helpful coding assistant"),
                 ChatMessage(role="user", content=prompt),
             ]
             resp = Anthropic(model=model).chat(response)
@@ -102,16 +104,18 @@ if uploaded_files:
 
                 # Generation
                 # resp = ollama.generate(model='codellama', 
-                #                     prompt=f"""You are a helpful coding assistant. Your task is to generate unit test cases for this function : {snippet}\
-                #                     \n\nPolitely refuse if the function is not suitable for generating test cases.
-                #                     \n\nGenerate atleast 5 unit test case. Include couple of edge cases as well.
-                #                     \n\nThere should be no duplicate test cases. Avoid generating repeated statements.
+                #                     prompt=f""" Your task is to generate unit test cases for this function : {snippet}\
+                #                     \n\n Politely refuse if the function is not suitable for generating test cases.
+                #                     \n\n Generate atleast 5 unit test case. Include couple of edge cases as well.
+                #                     \n\n There should be no duplicate test cases.
+                #                   \n\n Avoid generating repeated statements.
                 #                     """)
 
-                prompt=f"""You are a helpful coding assistant. Your task is to generate unit test cases for this function : \n\n{snippet}\
-                        \n\nGenerate 5 to 10 unit test cases. Include couple of edge cases as well.
-                        \n\nPolitely refuse if the function is not suitable for generating test cases.
-                        \n\nThere should be no duplicate test cases. Avoid generating repeated statements.
+                prompt=f""" Your task is to generate unit test cases for this function : \n\n{snippet}\
+                        \n\n Generate 5 to 10 unit test cases. Include couple of edge cases as well.
+                        \n\n Politely refuse if the function is not suitable for generating test cases.
+                        \n\n There should be no duplicate test cases. 
+                        \n\n Avoid generating repeated statements.
                     """
 
                 # print(prompt)