api keys

SermetPekin · Aug 7, 2024 · 8d2082d · 8d2082d
1 parent dd7cc4f
commit 8d2082d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -56,6 +56,16 @@ prompt = '''
 result, notes = chat(prompt, test=False) 
 print(result)
 ```
+## Defining Api Keys ( .env file )
+
+```bash
+  # .env file content 
+
+  OPENAI_API_KEY = "sk-proj-ABCDEFGIJKLMNOPQRSTUXVZ"
+  EVDS_API_KEY=ABCDEFGIJKLMNOP
+
+```
+
 
 
 ## New in this version 

diff --git a/evdschat/core/chat.py b/evdschat/core/chat.py
@@ -87,7 +87,11 @@ def chat_console():
     result, notes = chat(prompt=args.prompt, debug=args.debug, test=args.test)
     print(result)
     print(notes)
-    result.to_excel(f"{args.file_name}.xlsx")
+    def correct(x:str):
+        if x.endswith('.xlsx') : return x 
+        return f'{x}.xlsx'
+
+    result.to_excel(f"{correct(args.file_name)}")
 
 
 if __name__ == "__main__":