From 623cee8be4846e5762ff2949e519335ef8dee2bb Mon Sep 17 00:00:00 2001
From: suerprisePlus <15810472099@163.com>
Date: 星期一, 09 九月 2024 18:30:55 +0800
Subject: [PATCH] 大模型知识图谱接口对接

---
 src/main/java/com/yb/util/EntityHttpUtil.java |   36 ++++++++++++++++++++++++++----------
 1 files changed, 26 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/yb/util/EntityHttpUtil.java b/src/main/java/com/yb/util/EntityHttpUtil.java
index 6ca0733..11cea99 100644
--- a/src/main/java/com/yb/util/EntityHttpUtil.java
+++ b/src/main/java/com/yb/util/EntityHttpUtil.java
@@ -1,7 +1,9 @@
 package com.yb.util;
 
 import com.alibaba.fastjson.JSON;
+import com.yb.config.XzConfig;
 import okhttp3.*;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 
 import org.springframework.stereotype.Component;
@@ -16,23 +18,37 @@
 @Component
 @RestController
 public class EntityHttpUtil {
-
-    @Value("${spring.geographical.token}")
-    String token ;
-
-    @Value("${spring.geographical.url}")
-    String url ;
+    @Autowired
+    private XzConfig xzConfig;
 
     public  String getMessage(String aurl){
         try{
-            String axiosUrl =  url+aurl;
+            String axiosUrl = xzConfig.geographicalUrl +aurl;
             OkHttpClient client = new OkHttpClient.Builder().connectTimeout(5, TimeUnit.SECONDS)
                     .readTimeout(5, TimeUnit.SECONDS)
                     .writeTimeout(5, TimeUnit.SECONDS)
                     .build();
             Request request = new Request.Builder()
-                    .addHeader("Token",token)
+                    .addHeader("Token",xzConfig.token)
                     .url(axiosUrl)
+                    .build();
+            Response response = client.newCall(request).execute();
+            return response.body().string();
+        }catch (IOException e){
+            System.out.println(e.getMessage());
+        }
+        return  null;
+    }
+
+    public  String getAgentMessage(String aurl){
+        try{
+
+            OkHttpClient client = new OkHttpClient.Builder().connectTimeout(0, TimeUnit.SECONDS)
+                    .readTimeout(0, TimeUnit.SECONDS)
+                    .writeTimeout(0, TimeUnit.SECONDS)
+                    .build();
+            Request request = new Request.Builder()
+                    .url(aurl)
                     .build();
             Response response = client.newCall(request).execute();
             return response.body().string();
@@ -43,7 +59,7 @@
     }
     public String getPostMessage(String aurl, HashMap<String,String> parms){
         try{
-            String axiosUrl = url+aurl;
+            String axiosUrl = xzConfig.geographicalUrl+aurl;
             FormBody.Builder builder= new FormBody.Builder();
             for(String key:parms.keySet()){
                 builder.add(key,parms.get(key));
@@ -54,7 +70,7 @@
                     .writeTimeout(5, TimeUnit.SECONDS)
                     .build();
             Request request = new Request.Builder()
-                    .addHeader("Token",token)
+                    .addHeader("Token",xzConfig.token)
                     .url(axiosUrl)
                     .post(requestBody)
                     .build();

--
Gitblit v1.9.3