Commit 9e7cb993 authored by Mohamad Bashar Desoki's avatar Mohamad Bashar Desoki

Integrate with llama 3

parent 9b53736c
......@@ -25,6 +25,10 @@ public class Bot extends TelegramLongPollingBot {
var user = msg.getFrom();
var id = user.getId();
llama3 l3 = new llama3();
prompts = msg.getText();
String resp = l3.llmChat(msg.getText());
sendText(id,resp + " \n \uD83D\uDC50\uD83C\uDFFB back at ya! \uD83D\uDE0A");
System.out.println(user.getFirstName() + " wrote " + msg.getText());
}
......
package org.example;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
public class llama3 {
public String llmChat(String chat){
String prompt = "{\n" +
// " \"systemPrompt\": \" العب دور مساعد تقني يتحدث اللغة العربية بفصاحة \",\n" +
// " \"user\": \" مرحبا\",\n" +
// " \"Assistant\": \" أهلا، أسألني عن أحدث التقنيات لأجيبك\",\n" +
" \"prompt\": \""+ chat +"\"\n" +
"}";
String response = null;
response = sendRequestToAPI(prompt);
// System.out.println(response);
String jsonString = "";
jsonString = response.toString().substring(0,response.length()-51);
return jsonString;
}
public String sendRequestToAPI(String prompt) {
try {
URL url = new URL("https://fumes-api.onrender.com/llama3");
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
connection.setRequestMethod("POST");
connection.setRequestProperty("Content-Type", "application/json");
connection.setDoOutput(true);
try (OutputStream os = connection.getOutputStream()) {
byte[] input = prompt.getBytes("utf-8");
os.write(input, 0, input.length);
}
StringBuilder response = new StringBuilder();
try (BufferedReader br = new BufferedReader(new InputStreamReader(connection.getInputStream(), "utf-8"))) {
String responseLine = null;
while ((responseLine = br.readLine()) != null) {
response.append(responseLine.trim());
}
}
connection.disconnect();
String responseBody = response.toString();
// System.out.println(responseBody);
return responseBody;
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment