nit change print to logging

chenyuan99 · Apr 15, 2024 · 28d3dda · 28d3dda
1 parent ac06a01
commit 28d3dda
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 5 deletions.
diff --git a/jobgpt/generator.py b/jobgpt/generator.py
@@ -1,8 +1,8 @@
 import logging
-
+import os
 import openai
 
-openai.api_key = "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"
+openai.api_key = os.environ['OPENAI_API_KEY']
 
 
 def generate_response(prompt: str) -> str:

diff --git a/jobgpt/prompt.py b/jobgpt/prompt.py
@@ -1,3 +1,5 @@
 RESUME_COMMON = "Please resume the training with the following command:\n\n"
 
 REFERAL_COMMON = "Please refer to the following command:\n\n"
+
+COVER_LETTER_COMMON = "Please refer to the following command:\n\n"
diff --git a/jobgpt/tests.py b/jobgpt/tests.py
@@ -3,4 +3,8 @@
 from jobgpt.generator import generate_response
 
 # Create your tests here.
-# assert generate_response("hello")
+def test_generate_response():
+ try:
+ generate_response("hello")
+ finally:
+ assert True
diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,9 @@
-aiohttp
+aiohttp==3.9.2
 aiosignal==1.3.1
 asgiref==3.7.2
 async-timeout==4.0.2
 attrs==23.1.0
-certifi
+certifi==2024.2.2
 charset-normalizer==3.1.0
 defusedxml==0.7.1
 diff-match-patch==20230430