gperdrizet commited on
Commit
6ca7e38
·
verified ·
1 Parent(s): 8d4db90

Switched to llama3.1-8B-instruct for summarization and qwen2.5-coder-8B-instruct for agent

Browse files
Files changed (1) hide show
  1. configuration.py +14 -4
configuration.py CHANGED
@@ -11,20 +11,30 @@ DEFAULT_GITHUB_PROFILE = "https://github.com/gperdrizet"
11
  # max_tokens=8000
12
  # )
13
 
 
 
 
14
  SUMMARIZER_MODEL = OpenAI(
15
- base_url="https://gperdrizet--deepseek-R1-qwen-32B-serve.modal.run/v1",
16
  api_key=os.environ['MODAL_TOKEN_SECRET']
17
  )
18
 
 
 
 
 
 
 
 
19
  AGENT_MODEL = OpenAIServerModel(
20
- model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", # Same as HF model string
21
- api_base="https://gperdrizet--deepseek-R1-qwen-32B-serve.modal.run/v1",
22
  api_key=os.environ["MODAL_TOKEN_SECRET"],
23
  )
24
 
25
  INSTRUCTIONS = """
26
  You are an AI agent responsible for writing a resume based on the provided context. Your task is to generate a well-structured and professional resume that highlights the user's skills, experiences, and achievements.
27
- You will receive Two pieces of JSON structured context: a job call and a LinkedIn profile.
28
 
29
  LINKEDIN PROFILE EXAMPLE
30
 
 
11
  # max_tokens=8000
12
  # )
13
 
14
+ # Will be used for single shot summarization with no-frills prompting
15
+ # (e.g. job call extraction). It needs to output JSON formatted text,
16
+ # but this task does not require any complex reasoning or planning.
17
  SUMMARIZER_MODEL = OpenAI(
18
+ base_url="https://gperdrizet--llama-3-1-8b-instruct-serve.modal.run/v1",
19
  api_key=os.environ['MODAL_TOKEN_SECRET']
20
  )
21
 
22
+ # Will be used for resume resume writing agent via HuggingFace smolagents
23
+ # Including selection of relevant projects from GitHub profile
24
+ #
25
+ # Notes:
26
+ # - DeepSeek-R1-Distill-Qwen-32B does not seem to work well with smolagents,
27
+ # has trouble correctly formatting responses as code.
28
+
29
  AGENT_MODEL = OpenAIServerModel(
30
+ model_id="Qwen/Qwen2.5-Coder-14B-Instruct", # Same as HF model string
31
+ api_base="https://gperdrizet--qwen2-5-coder-14b-instruct-serve.modal.run/v1",
32
  api_key=os.environ["MODAL_TOKEN_SECRET"],
33
  )
34
 
35
  INSTRUCTIONS = """
36
  You are an AI agent responsible for writing a resume based on the provided context. Your task is to generate a well-structured and professional resume that highlights the user's skills, experiences, and achievements.
37
+ You will receive two pieces of JSON structured context: a job call and a LinkedIn profile.
38
 
39
  LINKEDIN PROFILE EXAMPLE
40