SerpAPI bug fixed (#630)

xebia-functional · Jan 9, 2024 · 180e829 · 180e829
1 parent 3acb85b
commit 180e829
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/examples/src/main/kotlin/com/xebia/functional/xef/conversation/reasoning/SerpApiExample.kt b/examples/src/main/kotlin/com/xebia/functional/xef/conversation/reasoning/SerpApiExample.kt
@@ -12,12 +12,12 @@ suspend fun main() {
 
   val answer = client.search(searchData)
 
-  answer.searchResults.forEach {
+  answer.searchResults?.forEach {
     println(
       "\n\uD83E\uDD16 Search Information:\n\n" +
         "Title: ${it.title}\n" +
         "Document: ${it.document}\n" +
         "Source: ${it.source}\n"
     )
-  }
+  } ?: println("No results found")
 }
diff --git a/reasoning/src/commonMain/kotlin/com/xebia/functional/xef/reasoning/serpapi/SearchResults.kt b/reasoning/src/commonMain/kotlin/com/xebia/functional/xef/reasoning/serpapi/SearchResults.kt
@@ -4,4 +4,4 @@ import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
 
 @Serializable
-data class SearchResults(@SerialName("organic_results") val searchResults: List<SearchResult>)
+data class SearchResults(@SerialName("organic_results") val searchResults: List<SearchResult>?)
diff --git a/reasoning/src/commonMain/kotlin/com/xebia/functional/xef/reasoning/serpapi/SearchTool.kt b/reasoning/src/commonMain/kotlin/com/xebia/functional/xef/reasoning/serpapi/SearchTool.kt
@@ -34,7 +34,7 @@ interface SearchTool : Tool {
         prompt =
           Prompt(model) {
             +system("Search results:")
-            docs.searchResults.take(maxResultsInContext).forEach {
+            docs.searchResults?.take(maxResultsInContext)?.forEach {
               +system("Title: ${it.title}")
               +system("Source: ${it.source}")
               +system("Content: ${it.document}")