Skip to content

Commit 2feada9

Browse files
committed
feat: add function
1 parent 325b494 commit 2feada9

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

backend/src/main/java/ch/xxx/aidoclibchat/adapter/config/FunctionConfig.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,19 +22,29 @@
2222
import org.springframework.context.annotation.Configuration;
2323

2424
import ch.xxx.aidoclibchat.domain.client.OpenLibraryClient;
25+
import ch.xxx.aidoclibchat.domain.client.TmdbClient;
2526

2627
@Configuration
2728
public class FunctionConfig {
2829
private final OpenLibraryClient openLibraryClient;
30+
private final TmdbClient tmdbClient;
2931
public static final String OPEN_LIBRARY_CLIENT = "openLibraryClient";
32+
public static final String THE_MOVIE_DATABASE_CLIENT = "theMovieDatabaseClient";
3033

31-
public FunctionConfig(OpenLibraryClient openLibraryClient) {
34+
public FunctionConfig(OpenLibraryClient openLibraryClient, TmdbClient tmdbClient) {
3235
this.openLibraryClient = openLibraryClient;
36+
this.tmdbClient = tmdbClient;
3337
}
3438

3539
@Bean(OPEN_LIBRARY_CLIENT)
3640
@Tool(description = "Search for books by author, title or subject.")
3741
public Function<OpenLibraryClient.Request, OpenLibraryClient.Response> openLibraryClient() {
3842
return this.openLibraryClient::apply;
3943
}
44+
45+
@Bean(THE_MOVIE_DATABASE_CLIENT)
46+
@Tool(description = "Search for movies by title.")
47+
public Function<TmdbClient.Request, TmdbClient.Response> theMovieDatabaseClient() {
48+
return this.tmdbClient::apply;
49+
}
4050
}

0 commit comments

Comments
 (0)