codewithdark commited on
Commit
d659715
·
verified ·
1 Parent(s): 11e1534

Create utils.py

Browse files
Files changed (1) hide show
  1. utility/utils.py +34 -0
utility/utils.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from datetime import datetime
3
+ import json
4
+
5
+ # Log types
6
+ LOG_TYPE_GPT = "GPT"
7
+ LOG_TYPE_PEXEL = "PEXEL"
8
+
9
+ # log directory paths
10
+ DIRECTORY_LOG_GPT = ".logs/gpt_logs"
11
+ DIRECTORY_LOG_PEXEL = ".logs/pexel_logs"
12
+
13
+ # method to log response from pexel and openai
14
+ def log_response(log_type, query,response):
15
+ log_entry = {
16
+ "query": query,
17
+ "response": response,
18
+ "timestamp": datetime.now().isoformat()
19
+ }
20
+ if log_type == LOG_TYPE_GPT:
21
+ if not os.path.exists(DIRECTORY_LOG_GPT):
22
+ os.makedirs(DIRECTORY_LOG_GPT)
23
+ filename = '{}_gpt3.txt'.format(datetime.now().strftime("%Y%m%d_%H%M%S"))
24
+ filepath = os.path.join(DIRECTORY_LOG_GPT, filename)
25
+ with open(filepath, "w") as outfile:
26
+ outfile.write(json.dumps(log_entry) + '\n')
27
+
28
+ if log_type == LOG_TYPE_PEXEL:
29
+ if not os.path.exists(DIRECTORY_LOG_PEXEL):
30
+ os.makedirs(DIRECTORY_LOG_PEXEL)
31
+ filename = '{}_pexel.txt'.format(datetime.now().strftime("%Y%m%d_%H%M%S"))
32
+ filepath = os.path.join(DIRECTORY_LOG_PEXEL, filename)
33
+ with open(filepath, "w") as outfile:
34
+ outfile.write(json.dumps(log_entry) + '\n')