package com.yb;/*
|
*@title OllamaApiTest
|
*@description
|
*@author yb
|
*@version 1.0
|
*@create 2024/5/28 21:12
|
*/
|
|
import org.apache.http.HttpEntity;
|
import org.apache.http.client.methods.CloseableHttpResponse;
|
import org.apache.http.client.methods.HttpPost;
|
import org.apache.http.entity.StringEntity;
|
import org.apache.http.impl.client.CloseableHttpClient;
|
import org.apache.http.impl.client.HttpClients;
|
import org.apache.http.util.EntityUtils;
|
import org.json.JSONObject;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.boot.test.context.SpringBootTest;
|
|
import java.nio.charset.Charset;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
|
@SpringBootTest
|
public class OllamaApiTest {
|
|
|
@Test
|
public void test() throws Exception{
|
CloseableHttpClient client = HttpClients.createDefault();
|
//ollama 默认端口是 11434
|
HttpPost post = new HttpPost("http://192.168.11.104:11434/api/generate");
|
post.addHeader("Content-type","application/json; charset=utf-8");
|
//参数
|
Map<String, Object> map = new HashMap<>();
|
map.put("model","llama3:8b");
|
map.put("prompt","你好");
|
//不以流式返回
|
map.put("stream",false);
|
|
StringEntity stringEntity = new StringEntity(JSONObject.valueToString(map),Charset.forName("UTF-8"));
|
post.setEntity(stringEntity);
|
CloseableHttpResponse response = client.execute(post);
|
HttpEntity entity = response.getEntity();
|
System.out.println(EntityUtils.toString(entity));
|
}
|
|
}
|
