Skip to content

Commit 303c318

Browse files
committed
chore: allow overriding LLM model in mergeMembersWithLLM workflow
Signed-off-by: Joana Maia <jmaia@contractor.linuxfoundation.org>
1 parent c08636f commit 303c318

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

services/apps/merge_suggestions_worker/src/types.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@ export interface IProcessMergeOrganizationSuggestionsWithLLM {
104104
export interface IProcessMergeMemberSuggestionsWithLLM {
105105
similarity: ISimilarityFilter
106106
tenantId: string
107+
modelId?: string
107108
}
108109

109110
export interface ISimilarMemberOpensearchResult {

services/apps/merge_suggestions_worker/src/workflows/mergeMembersWithLLM.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { continueAsNew, proxyActivities } from '@temporalio/workflow'
22

3-
import { LLMSuggestionVerdictType, MemberMergeSuggestionTable } from '@crowd/types'
3+
import { LLMSuggestionVerdictType, LlmModelType, MemberMergeSuggestionTable } from '@crowd/types'
44

55
import * as commonActivities from '../activities/common'
66
import * as memberActivities from '../activities/memberMergeSuggestions'
@@ -24,7 +24,7 @@ export async function mergeMembersWithLLM(
2424
): Promise<void> {
2525
const SUGGESTIONS_PER_RUN = 10
2626
const REGION = 'us-east-1'
27-
const MODEL_ID = 'us.anthropic.claude-sonnet-4-20250514-v1:0'
27+
const MODEL_ID = args.modelId ?? LlmModelType.CLAUDE_SONNET_4
2828
const MODEL_ARGS = {
2929
max_tokens: 2000,
3030
anthropic_version: 'bedrock-2023-05-31',

0 commit comments

Comments
 (0)