Fetch ranker relevance scores from GraphQL (#5066)

Use https://github.com/sourcegraph/sourcegraph/pull/64172 in Cody ## Test plan - tested locally - code behind a feature flag
sourcegraph · Jul 31, 2024 · a81c35a · a81c35a
1 parent d5c6c1f
commit a81c35a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/lib/shared/src/sourcegraph-api/graphql/client.ts b/lib/shared/src/sourcegraph-api/graphql/client.ts
@@ -332,8 +332,8 @@ type RecordContextResponse = unknown
 interface RankContextResponse {
     rankContext: {
         ranker: string
-        used: number[]
-        ignored: number[]
+        used: { index: number; score: number }[]
+        ignored: { index: number; score: number }[]
     }
 }
 

diff --git a/lib/shared/src/sourcegraph-api/graphql/queries.ts b/lib/shared/src/sourcegraph-api/graphql/queries.ts
@@ -213,8 +213,14 @@ export const RANK_CONTEXT_QUERY = `
 query RankContext($interactionId: String!, $query: String!, $contextItems: [InputContextItem!]!) {
     rankContext(interactionId: $interactionId, query:$query, contextItems: $contextItems) {
         ranker
-        used
-        ignored
+        used {
+            index
+            score
+        }
+        ignored {
+            index
+            score
+        }
     }
 }`
 

diff --git a/vscode/src/chat/context/contextAPIClient.ts b/vscode/src/chat/context/contextAPIClient.ts
@@ -18,7 +18,11 @@ function toInput(input: ContextItem[]): InputContextItem[] {
                       retriever: i.source || '',
                   }
         )
-        .filter(i => i !== null) as InputContextItem[]
+        .filter(notNull)
+}
+
+function notNull<T>(value: T | null | undefined): value is T {
+    return value !== null && value !== undefined
 }
 
 export class ContextAPIClient {