Skip to content

Commit edeaa12

Browse files
committed
Fix unit test
1 parent 601cdc8 commit edeaa12

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

exec/java-exec/src/main/resources/webapp/src/hooks/useProspector.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,7 @@ export function useProspector(
367367
console.error(`Tool "${toolCall.name}" failed`, err);
368368
return JSON.stringify({ error: msg });
369369
}
370-
}, [onSqlGenerated, onVisualizationCreated]);
370+
}, [onSqlGenerated, onVisualizationCreated, sendDataToAiRef]);
371371

372372
const doStreamRound = useCallback((
373373
allMessages: ChatMessage[],

exec/java-exec/src/test/java/org/apache/drill/exec/server/rest/ai/HttpClientFactoryTest.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
package org.apache.drill.exec.server.rest.ai;
1919

2020
import okhttp3.OkHttpClient;
21+
import okhttp3.Proxy;
2122
import org.junit.jupiter.api.Test;
2223

2324
import java.net.InetSocketAddress;
@@ -133,7 +134,7 @@ public void testNoProxyConfiguration() throws Exception {
133134
config.setProxyUrl(null);
134135

135136
OkHttpClient client = HttpClientFactory.createClient(config);
136-
assertNull(client.proxy());
137+
assertEquals(Proxy.Type.DIRECT, client.proxy().type());
137138
}
138139

139140
@Test
@@ -142,7 +143,7 @@ public void testEmptyProxyConfiguration() throws Exception {
142143
config.setProxyUrl("");
143144

144145
OkHttpClient client = HttpClientFactory.createClient(config);
145-
assertNull(client.proxy());
146+
assertEquals(Proxy.Type.DIRECT, client.proxy().type());
146147
}
147148

148149
@Test

0 commit comments

Comments
 (0)