Updated ask and askAsync responses to include responseTime parameter

This commit is contained in:
Amith Koujalgi 2023-11-16 00:50:56 +05:30
parent 79f69c1620
commit a5c6ba5558

View File

@ -194,7 +194,7 @@ public class OllamaAPI {
* *
* @param ollamaModelType the ollama model to ask the question to * @param ollamaModelType the ollama model to ask the question to
* @param promptText the prompt/question text * @param promptText the prompt/question text
* @return the response text from the model * @return OllamaResult - that includes response text and time taken for response
*/ */
public OllamaResult ask(String ollamaModelType, String promptText) throws OllamaBaseException, IOException, InterruptedException { public OllamaResult ask(String ollamaModelType, String promptText) throws OllamaBaseException, IOException, InterruptedException {
OllamaRequestModel ollamaRequestModel = new OllamaRequestModel(ollamaModelType, promptText); OllamaRequestModel ollamaRequestModel = new OllamaRequestModel(ollamaModelType, promptText);