This commit is contained in:
Amith Koujalgi
2023-11-07 23:06:02 +05:30
parent 153f516d9f
commit c74e70f3fa
2 changed files with 7 additions and 2 deletions

View File

@@ -10,7 +10,7 @@ public class Main {
OllamaAPI ollamaAPI = new OllamaAPI(host);
String prompt1 = SamplePrompts.getSampleDatabasePromptWithQuestion("List all customer names who have bought one or more products");
String response1 = ollamaAPI.ask(OllamaModelType.LLAMA2, prompt1);
String response1 = ollamaAPI.ask(OllamaModelType.SQLCODER, prompt1);
System.out.println(response1);
String prompt2 = "Give me a list of world cup cricket teams.";