Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -31,11 +31,11 @@ export async function mergeMembersWithLLM(
): Promise<void> {
const SUGGESTIONS_PER_RUN = 10
const REGION = 'us-east-1'
const MODEL_ID = 'us.anthropic.claude-sonnet-4-20250514-v1:0'
const MODEL_ID = 'us.anthropic.claude-sonnet-5'
const MODEL_ARGS = {
max_tokens: 2000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'disabled' },
}
const PROMPT = `Please compare and decide if these two members are the same person or not.
Only compare data from first member and second member. Never compare data from only one member with itself.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,11 +32,11 @@ export async function mergeOrganizationsWithLLM(
): Promise<void> {
const SUGGESTIONS_PER_RUN = 5
const REGION = 'us-east-1'
const MODEL_ID = 'us.anthropic.claude-sonnet-4-20250514-v1:0'
const MODEL_ID = 'us.anthropic.claude-sonnet-5'
const MODEL_ARGS = {
max_tokens: 2000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'disabled' },
}
const PROMPT = `Please compare and decide if these two organizations are the same organization or not. Submit your answer with the submit_merge_decision tool: set decision to true if they are the same organization, false otherwise, and give the reason in one short sentence.`

Expand Down
1 change: 1 addition & 0 deletions services/libs/types/src/enums/llm.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ export enum LlmModelType {
CLAUDE_3_5_SONNET_V2 = 'anthropic.claude-3-5-sonnet-20241022-v2:0',
CLAUDE_3_OPUS = 'anthropic.claude-3-opus-20240229-v1:0',
CLAUDE_SONNET_4 = 'us.anthropic.claude-sonnet-4-20250514-v1:0',
CLAUDE_SONNET_5 = 'us.anthropic.claude-sonnet-5',
CLAUDE_SONNET_5_5 = 'us.anthropic.claude-sonnet-5-5',
CLAUDE_HAIKU_4_5 = 'us.anthropic.claude-haiku-4-5-20251001-v1:0',
}
Expand Down
33 changes: 19 additions & 14 deletions services/libs/types/src/llm.ts
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@ export const LLM_MODEL_REGION_MAP: Record<LlmModelType, string> = {
[LlmModelType.CLAUDE_3_5_SONNET]: 'us-east-1',
[LlmModelType.CLAUDE_3_5_SONNET_V2]: 'us-west-2',
[LlmModelType.CLAUDE_SONNET_4]: 'us-east-1',
[LlmModelType.CLAUDE_SONNET_5]: 'us-east-1',
[LlmModelType.CLAUDE_SONNET_5_5]: 'us-east-1',
[LlmModelType.CLAUDE_HAIKU_4_5]: 'us-east-1',
}
Expand All @@ -56,6 +57,10 @@ export const LLM_MODEL_PRICING_MAP: Record<LlmModelType, ILlmPricing> = {
costPer1000InputTokens: 0.003,
costPer1000OutputTokens: 0.015,
},
[LlmModelType.CLAUDE_SONNET_5]: {
costPer1000InputTokens: 0.002,
costPer1000OutputTokens: 0.01,
},
[LlmModelType.CLAUDE_SONNET_5_5]: {
costPer1000InputTokens: 0.002,
costPer1000OutputTokens: 0.01,
Expand Down Expand Up @@ -84,11 +89,11 @@ export function estimateLlmCostUsd(

export const LLM_SETTINGS: Record<LlmQueryType, ILlmSettings> = {
[LlmQueryType.MEMBER_ENRICHMENT]: {
modelId: LlmModelType.CLAUDE_SONNET_4,
modelId: LlmModelType.CLAUDE_SONNET_5_5,
arguments: {
max_tokens: 65536,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'between_tools' },
},
},
[LlmQueryType.MEMBER_ENRICHMENT_FIND_RELATED_LINKEDIN_PROFILES]: {
Expand All @@ -100,19 +105,19 @@ export const LLM_SETTINGS: Record<LlmQueryType, ILlmSettings> = {
},
},
[LlmQueryType.MEMBER_ENRICHMENT_SQUASH_MULTIPLE_VALUE_ATTRIBUTES]: {
modelId: LlmModelType.CLAUDE_SONNET_4,
modelId: LlmModelType.CLAUDE_SONNET_5_5,
arguments: {
max_tokens: 2000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'between_tools' },
},
},
[LlmQueryType.MEMBER_ENRICHMENT_SQUASH_WORK_EXPERIENCES_FROM_MULTIPLE_SOURCES]: {
modelId: LlmModelType.CLAUDE_SONNET_4,
modelId: LlmModelType.CLAUDE_SONNET_5_5,
arguments: {
max_tokens: 8000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'between_tools' },
},
},
[LlmQueryType.MATCH_MAIN_GITHUB_ORGANIZATION_AND_DESCRIPTION]: {
Expand All @@ -124,19 +129,19 @@ export const LLM_SETTINGS: Record<LlmQueryType, ILlmSettings> = {
},
},
[LlmQueryType.REPO_CATEGORIES]: {
modelId: LlmModelType.CLAUDE_SONNET_4,
modelId: LlmModelType.CLAUDE_SONNET_5_5,
arguments: {
max_tokens: 2000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'between_tools' },
},
},
[LlmQueryType.REPO_COLLECTIONS]: {
modelId: LlmModelType.CLAUDE_SONNET_4,
modelId: LlmModelType.CLAUDE_SONNET_5_5,
arguments: {
max_tokens: 2000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'between_tools' },
},
},
[LlmQueryType.MEMBER_BOT_VALIDATION]: {
Expand All @@ -148,19 +153,19 @@ export const LLM_SETTINGS: Record<LlmQueryType, ILlmSettings> = {
},
},
[LlmQueryType.SELECT_MOST_RELEVANT_DOMAIN]: {
modelId: LlmModelType.CLAUDE_SONNET_4,
modelId: LlmModelType.CLAUDE_SONNET_5_5,
arguments: {
max_tokens: 2000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'between_tools' },
},
},
[LlmQueryType.FAKE_ORGANIZATION_ANALYSIS]: {
modelId: LlmModelType.CLAUDE_SONNET_4,
modelId: LlmModelType.CLAUDE_SONNET_5_5,
arguments: {
max_tokens: 2000,
anthropic_version: 'bedrock-2023-05-31',
temperature: 0,
thinking: { type: 'between_tools' },
},
},
[LlmQueryType.PROJECT_EVALUATION]: {
Expand Down
Loading