bug fixes, global personality works

Author: bachittle

filegpt/FileGPT.py

@@ -7,11 +7,13 @@
 import tempfile
 
 from openai_wrapper import get_openai_response
-from personalities import PERSONALITIES, get_personality
+from personalities import PERSONALITIES, get_personality, get_openai_personality
 
 import tiktoken
 from typing import List
 
+PERSONALITY = None
+
 MODELS = [
     "gpt-3.5-turbo",
     "gpt-4"
@@ -38,7 +40,9 @@ def read_input(files: List[str] = None) -> str:
         content = "\n".join(contents)
     else:
         logging.info("Reading input from stdin")
-        print("Reading from stdin (press CTRL+D for linux/mac or Enter+CTRL+Z+Enter for windows to stop)...")
+        if os.isatty(sys.stdin.fileno()):
+            print("Current personality:", get_personality(PERSONALITY))
+            print("Reading from stdin (press CTRL+D for linux/mac or Enter+CTRL+Z+Enter for windows to stop)...")
         content = sys.stdin.read()
     return content
 
@@ -80,7 +84,8 @@ def process_text(model_name: str, input_files: List[str] = None) -> str:
 
     logging.info("Input has %d tokens", len(enc.encode(input_content)))
 
-    response = get_openai_response(model_name, [{"role": "user", "content": input_content}])
+    print(get_openai_personality(PERSONALITY))
+    response = get_openai_response(model_name, [get_openai_personality(PERSONALITY), {"role": "user", "content": input_content}])
     resp_str = write_output(response)
     logging.info("FileGPT finished, response has %d tokens", len(enc.encode(resp_str)))
 
@@ -102,8 +107,9 @@ def main():
             # list available personalities
             print(f"Available personalities: {', '.join(PERSONALITIES)}")
             return
-
 
+    global PERSONALITY
+    PERSONALITY = get_personality(args.personality)
     process_text(args.model, args.file)
 
 

filegpt/personalities.py

@@ -1,23 +1,34 @@
 PERSONALITIES = [
-    "python_coder",
+    "coder",
 ]
 
 MAP_PERSONALITY_TO_OPENAI = {
-    "python_coder": """
+    "coder": {"role": "system", "content": """
 You only give your responses with python code.    
 When you need to tell the user something, use code comments.
 
 Example:
 User: What is your name?
 Bot: # My name is GPT-3
-User: Write some code to print your name
-Bot: print("GPT-3")
-"""
+User: Write some code that performs the fibonacci sequence
+Bot: 
+# Sure! Here you go:
+def fibonacci(n):
+...
+
+Do not use markdown or html in your responses.
+Do not wrap your response in backticks (```python)
+"""}
 }
 
 def get_personality(personality: str) -> str:
     if personality in PERSONALITIES:
         return personality
     else:
-        print(f"Personality {personality} not found. Available personalities: {', '.join(PERSONALITIES)}")
-        return None 
+        return None 
+
+def get_openai_personality(personality: str) -> str:
+    if personality in MAP_PERSONALITY_TO_OPENAI:
+        return MAP_PERSONALITY_TO_OPENAI[personality]
+    else:
+        return None