Loading chatmol.py +3 −2 Original line number Original line Diff line number Diff line Loading @@ -7,6 +7,7 @@ from pymol import cmd import http.server import http.server class PyMOLCommandHandler(http.server.BaseHTTPRequestHandler): class PyMOLCommandHandler(http.server.BaseHTTPRequestHandler): def __init__(self): def __init__(self): from http import HTTPStatus from http import HTTPStatus Loading Loading @@ -74,7 +75,7 @@ stashed_commands = [] # Save API Key in ~/.PyMOL/apikey.txt # Save API Key in ~/.PyMOL/apikey.txt API_KEY_FILE = os.path.expanduser('~')+"/.PyMOL/apikey.txt" API_KEY_FILE = os.path.expanduser('~')+"/.PyMOL/apikey.txt" OPENAI_KEY_ENV = "OPENAI_API_KEY" OPENAI_KEY_ENV = "OPENAI_API_KEY" GPT_MODEL = "gpt-3.5-turbo" GPT_MODEL = "gpt-4o" client = None client = None def set_api_key(api_key): def set_api_key(api_key): Loading Loading @@ -109,7 +110,7 @@ def load_api_key(): def update_model(model_name): def update_model(model_name): global GPT_MODEL global GPT_MODEL GPT_MODEL = mdoel_name GPT_MODEL = model_name print("Model updated to: ", GPT_MODEL) print("Model updated to: ", GPT_MODEL) return "Model updated to: " + GPT_MODEL return "Model updated to: " + GPT_MODEL Loading chatmol_pkg/chatmol/utils.py +2 −2 Original line number Original line Diff line number Diff line Loading @@ -8,11 +8,11 @@ class ChatMol: def __init__(self, def __init__(self, openai_api_key=None, openai_api_key=None, verbose=False, verbose=False, gpt_model="gpt-3.5-turbo-1106", gpt_model="gpt-4o", chatgpt_max_history=10, chatgpt_max_history=10, chatgpt_temp=0, chatgpt_temp=0, chatgpt_max_tokens=256, chatgpt_max_tokens=256, claude_model="claude-3-opus-20240229", claude_model="claude-3-5-sonnet-20240620", in_pymol=False in_pymol=False ): ): self.in_pymol = in_pymol self.in_pymol = in_pymol Loading Loading
chatmol.py +3 −2 Original line number Original line Diff line number Diff line Loading @@ -7,6 +7,7 @@ from pymol import cmd import http.server import http.server class PyMOLCommandHandler(http.server.BaseHTTPRequestHandler): class PyMOLCommandHandler(http.server.BaseHTTPRequestHandler): def __init__(self): def __init__(self): from http import HTTPStatus from http import HTTPStatus Loading Loading @@ -74,7 +75,7 @@ stashed_commands = [] # Save API Key in ~/.PyMOL/apikey.txt # Save API Key in ~/.PyMOL/apikey.txt API_KEY_FILE = os.path.expanduser('~')+"/.PyMOL/apikey.txt" API_KEY_FILE = os.path.expanduser('~')+"/.PyMOL/apikey.txt" OPENAI_KEY_ENV = "OPENAI_API_KEY" OPENAI_KEY_ENV = "OPENAI_API_KEY" GPT_MODEL = "gpt-3.5-turbo" GPT_MODEL = "gpt-4o" client = None client = None def set_api_key(api_key): def set_api_key(api_key): Loading Loading @@ -109,7 +110,7 @@ def load_api_key(): def update_model(model_name): def update_model(model_name): global GPT_MODEL global GPT_MODEL GPT_MODEL = mdoel_name GPT_MODEL = model_name print("Model updated to: ", GPT_MODEL) print("Model updated to: ", GPT_MODEL) return "Model updated to: " + GPT_MODEL return "Model updated to: " + GPT_MODEL Loading
chatmol_pkg/chatmol/utils.py +2 −2 Original line number Original line Diff line number Diff line Loading @@ -8,11 +8,11 @@ class ChatMol: def __init__(self, def __init__(self, openai_api_key=None, openai_api_key=None, verbose=False, verbose=False, gpt_model="gpt-3.5-turbo-1106", gpt_model="gpt-4o", chatgpt_max_history=10, chatgpt_max_history=10, chatgpt_temp=0, chatgpt_temp=0, chatgpt_max_tokens=256, chatgpt_max_tokens=256, claude_model="claude-3-opus-20240229", claude_model="claude-3-5-sonnet-20240620", in_pymol=False in_pymol=False ): ): self.in_pymol = in_pymol self.in_pymol = in_pymol Loading
