Merge branch 'main' into open-ai-typed-models

xebia-functional · Jun 30, 2023 · c84d3ef · c84d3ef
2 parents f48bca3 + 6b4b66a
commit c84d3ef
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/java/src/main/java/com/xebia/functional/xef/java/auto/AIScope.java b/java/src/main/java/com/xebia/functional/xef/java/auto/AIScope.java
@@ -107,6 +107,25 @@ public CompletableFuture<List<String>> promptMessage(String prompt, LLM.Chat llm
         return future(continuation -> scope.promptMessage(prompt, llmModel, functions, user, echo, n, temperature, bringFromContext, minResponseTokens, continuation));
     }
 
+    public CompletableFuture<Void> extendContext(String[] docs) {
+        return future(continuation -> scope.extendContext(docs, continuation))
+                .thenApply(unit -> null);
+    }
+
+    public <A> CompletableFuture<A> contextScope(Function1<AIScope, CompletableFuture<A>> f) {
+        return future(continuation -> scope.contextScope((coreAIScope, continuation1) -> {
+            AIScope nestedScope = new AIScope(coreAIScope, AIScope.this);
+            return FutureKt.await(f.invoke(nestedScope), continuation);
+        }, continuation));
+    }
+
+    public <A> CompletableFuture<A> contextScope(VectorStore store, Function1<AIScope, CompletableFuture<A>> f) {
+        return future(continuation -> scope.contextScope(store, (coreAIScope, continuation1) -> {
+            AIScope nestedScope = new AIScope(coreAIScope, AIScope.this);
+            return FutureKt.await(f.invoke(nestedScope), continuation);
+        }, continuation));
+    }
+
     public <A> CompletableFuture<A> contextScope(List<String> docs, Function1<AIScope, CompletableFuture<A>> f) {
         return future(continuation -> scope.contextScopeWithDocs(docs, (coreAIScope, continuation1) -> {
             AIScope nestedScope = new AIScope(coreAIScope, AIScope.this);