made changes

Code+Folder/src/constants.py

@@ -6,7 +6,28 @@ OUTPUT_PATH = "./output/"
 JD_EMBEDDINGS_FILENAME = "jd_embeddings_large.pkl"
 RESUME_EMBEDDINGS_FILENAME = "resume_embeddings_large.pkl"
 IS_EMBEDDINGS_CREATED = True
-OPENAI_MODEL_NAME = "gpt-4o-mini-2024-07-18"
+PROVIDERS = {
+    "OpenAI": {
+        "env_var": "OPENAI_API_KEY",
+        "models": ["gpt-4o", "gpt-4o-mini", "gpt-4.1", "gpt-4.1-mini"],
+        "url": "https://platform.openai.com/api-keys",
+    },
+    "xAI": {
+        "env_var": "XAI_API_KEY",
+        "models": ["xai/grok-2-latest", "xai/grok-2-mini"],
+        "url": "https://console.x.ai/",
+    },
+    "Anthropic": {
+        "env_var": "ANTHROPIC_API_KEY",
+        "models": ["anthropic/claude-sonnet-4-5-20250929", "anthropic/claude-haiku-4-5-20251001"],
+        "url": "https://console.anthropic.com/settings/keys",
+    },
+    "Google": {
+        "env_var": "GEMINI_API_KEY",
+        "models": ["gemini/gemini-2.0-flash", "gemini/gemini-2.5-flash-preview-04-17"],
+        "url": "https://aistudio.google.com/apikey",
+    },
+}
 TEMPLATE_CONTENT = """You are a helpful assistant. You do not respond as 'User' or pretend to be 'User'. You only 
 respond once as 'assistant'. 
 
@@ -37,5 +58,3 @@ relevant_skills_highlight_prompt = "Analyze this resume: <RESUME STARTS HERE> {}
 resume_formatting_prompt = "Offer guidance on how the candidate can enhance the formatting of their resume: <RESUME STARTS HERE> {} <RESUME ENDS HERE> to improve visual appeal and readability."
 resume_length_prompt = "Recommend strategies for the candidate to adjust the length of their resume: <RESUME STARTS HERE> {} <RESUME ENDS HERE>, ensuring it is concise while remaining aligned with the requirements in the job description: <JOB DESCRIPTION STARTS HERE> {}. <JOB DESCRIPTION ENDS HERE>"
 
-# OPENAI_API_KEY is used but not defined in this file
-# OPENAI_API_KEY = "your-openai-api-key-here"  # Replace with your actual API key

Code+Folder/src/resume_suggestions.py

@@ -16,8 +16,8 @@ from langchain_classic.chains import ConversationChain
 from langchain_classic.memory import ConversationBufferWindowMemory
 from langchain_core.messages import SystemMessage
 from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
-from langchain_openai import ChatOpenAI
-from constants import OPENAI_MODEL_NAME, TEMPLATE_CONTENT, comparison_prompt, resume_analysis_prompt, \
+from langchain_community.chat_models import ChatLiteLLM
+from constants import PROVIDERS, TEMPLATE_CONTENT, comparison_prompt, resume_analysis_prompt, \
     job_description_analysis_prompt, gap_analysis_prompt, actionable_steps_prompt, experience_enhancement_prompt, \
     additional_qualifications_prompt, resume_tailoring_prompt, relevant_skills_highlight_prompt, \
     resume_formatting_prompt, resume_length_prompt
@@ -38,17 +38,26 @@ with st.sidebar:
     st.title('Resume Reviewer')
     st.write("Upload your resume for my recommendations. Job description is optional.")
 
-    # API Key
+    # Provider & Model selection
     st.write("---")
-    st.write("### OpenAI API Key")
-    st.write("Enter your API key to use the application. Get one at [OpenAI](https://platform.openai.com/api-keys)")
-    api_key = st.text_input("API Key", type="password", help="Your API key will not be stored")
+    st.write("### LLM Provider")
+    provider_names = list(PROVIDERS.keys())
+    selected_provider = st.selectbox("Provider", provider_names)
+    provider_info = PROVIDERS[selected_provider]
+
+    selected_model = st.selectbox("Model", provider_info["models"])
+
+    st.write(f"Get an API key at [{selected_provider}]({provider_info['url']})")
+    api_key = st.text_input(
+        f"{selected_provider} API Key", type="password",
+        help="Your API key will not be stored",
+    )
 
     if api_key:
-        os.environ['OPENAI_API_KEY'] = api_key
-        llm = ChatOpenAI(temperature=0.0, model=OPENAI_MODEL_NAME)
+        os.environ[provider_info["env_var"]] = api_key
+        llm = ChatLiteLLM(model=selected_model, temperature=0.0)
     else:
-        st.info("Please enter your OpenAI API key to start")
+        st.info(f"Please enter your {selected_provider} API key to start")
         llm = None
 
     # Resume upload (file only)
@@ -203,7 +212,7 @@ if llm is not None:
 
 def generate_response(prompt_input):
     if resume_chain is None:
-        return "Please enter your OpenAI API key to use this application"
+        return "Please enter your API key to use this application"
     output = resume_chain.predict(input=prompt_input)
     return output
 

app.py

@@ -16,8 +16,8 @@ from langchain_classic.chains import ConversationChain
 from langchain_classic.memory import ConversationBufferWindowMemory
 from langchain_core.messages import SystemMessage
 from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
-from langchain_openai import ChatOpenAI
-from constants import OPENAI_MODEL_NAME, TEMPLATE_CONTENT, comparison_prompt, resume_analysis_prompt, \
+from langchain_community.chat_models import ChatLiteLLM
+from constants import PROVIDERS, TEMPLATE_CONTENT, comparison_prompt, resume_analysis_prompt, \
     job_description_analysis_prompt, gap_analysis_prompt, actionable_steps_prompt, experience_enhancement_prompt, \
     additional_qualifications_prompt, resume_tailoring_prompt, relevant_skills_highlight_prompt, \
     resume_formatting_prompt, resume_length_prompt
@@ -38,17 +38,26 @@ with st.sidebar:
     st.title('Resume Reviewer')
     st.write("Upload your resume for my recommendations. Job description is optional.")
 
-    # API Key
+    # Provider & Model selection
     st.write("---")
-    st.write("### OpenAI API Key")
-    st.write("Enter your API key to use the application. Get one at [OpenAI](https://platform.openai.com/api-keys)")
-    api_key = st.text_input("API Key", type="password", help="Your API key will not be stored")
+    st.write("### LLM Provider")
+    provider_names = list(PROVIDERS.keys())
+    selected_provider = st.selectbox("Provider", provider_names)
+    provider_info = PROVIDERS[selected_provider]
+
+    selected_model = st.selectbox("Model", provider_info["models"])
+
+    st.write(f"Get an API key at [{selected_provider}]({provider_info['url']})")
+    api_key = st.text_input(
+        f"{selected_provider} API Key", type="password",
+        help="Your API key will not be stored",
+    )
 
     if api_key:
-        os.environ['OPENAI_API_KEY'] = api_key
-        llm = ChatOpenAI(temperature=0.0, model=OPENAI_MODEL_NAME)
+        os.environ[provider_info["env_var"]] = api_key
+        llm = ChatLiteLLM(model=selected_model, temperature=0.0)
     else:
-        st.info("Please enter your OpenAI API key to start")
+        st.info(f"Please enter your {selected_provider} API key to start")
         llm = None
 
     # Resume upload (file only)
@@ -192,7 +201,7 @@ if llm is not None:
 
 def generate_response(prompt_input):
     if resume_chain is None:
-        return "Please enter your OpenAI API key to use this application"
+        return "Please enter your API key to use this application"
     output = resume_chain.predict(input=prompt_input)
     return output
 

pyproject.toml

@@ -9,8 +9,7 @@ dependencies = [
     "langchain>=0.3.0",
     "langchain-community>=0.3.0",
     "langchain-core>=0.3.0",
-    "langchain-openai>=0.2.0",
-    "openai>=1.10.0",
+    "litellm>=1.0.0",
     "streamlit-feedback>=0.1.3",
     "PyPDF2>=3.0.1",
     "pypdf>=3.17.1",

requirements.txt

@@ -3,8 +3,7 @@ langchain>=0.3.0
 langchain-classic>=1.0.0
 langchain-community>=0.3.0
 langchain-core>=0.3.0
-langchain-openai>=0.2.0
-openai>=1.10.0
+litellm>=1.0.0
 streamlit-feedback>=0.1.3
 PyPDF2>=3.0.1
 pypdf>=3.17.1