V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
爱意满满的作品展示区。
wind1986
V2EX  ›  分享创造

Java 接入 Gemini SSE

  •  
  •   wind1986 · 343 天前 · 678 次点击
    这是一个创建于 343 天前的主题,其中的信息可能已经有所发展或是发生改变。

    有人成功的么?有点奇怪, 其他家的 SSE 都能正常接入, 就是 Gemini 的不行, 通过 okhttp 接入的 ,NodeJs 和 Python 之类均可以.
    调试之后发现返回 http 的头不对, application/json; charset=UTF-8, 手动更改为 text/event-stream, 不抛异常, 但是不能进入具体事件代码, 有人知道为什么么?
    随便找的一份代码

    @Slf4j
    @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
    public class SseTest {
    
        @Test
        public void test01() {
            // OkHttp 客户端
            OkHttpClient okHttpClient = new OkHttpClient.Builder()
                    .connectTimeout(10, TimeUnit.SECONDS)
                    .writeTimeout(50, TimeUnit.SECONDS)
                    .readTimeout(50, TimeUnit.SECONDS)
                    .build();
    
            // 事件源工厂
            EventSource.Factory factory = EventSources.createFactory(okHttpClient);
    
            // 请求体
            String requestBody = "{ \"contents\":[\n" +
                    "            {\"role\": \"user\",\n" +
                    "              \"parts\":[{\"text\": \"Write a story about a magic backpack.\"}]\n" +
                    "            }\n" +
                    "          ]\n" +
                    "        }";
    
            // 请求对象
            Request request = new Request.Builder()
                    .url("http://wainao.api-v2.cloudos.com/google/v1beta/models/gemini-pro:streamGenerateContent?key=AIzaSyBGJbctKS06Zn")
                    .post(RequestBody.create(MediaType.parse(ContentType.JSON.getValue()), requestBody))
                    .build();
    
            // 自定义监听器
            EventSourceListener eventSourceListener = new ConsoleEventSourceListener();
    
            // 创建事件
            EventSource eventSource = factory.newEventSource(request, eventSourceListener);
    
            // 等待线程结束
            CountDownLatch countDownLatch = new CountDownLatch(1);
            try {
                countDownLatch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    
    }
    
    @Slf4j
    class ConsoleEventSourceListener extends EventSourceListener {
    
        @Override
        public void onOpen(EventSource eventSource, Response response) {
            log.info("OpenAI 建立 sse 连接...");
        }
    
        @Override
        public void onEvent(EventSource eventSource, String id, String type, String data) {
            log.info("OpenAI 返回数据:{}", data);
            if ("[DONE]".equals(data)) {
                log.info("OpenAI 返回数据结束了");
                return;
            }
        }
    
        @Override
        public void onClosed(EventSource eventSource) {
            log.info("OpenAI 关闭 sse 连接...");
        }
    
        @SneakyThrows
        @Override
        public void onFailure(EventSource eventSource, Throwable t, Response response) {
            if (Objects.isNull(response)) {
                log.error("OpenAI  sse 连接异常:{}", t);
                eventSource.cancel();
                return;
            }
            ResponseBody body = response.body();
            if (Objects.nonNull(body)) {
                log.error("OpenAI  sse 连接异常 data:{},异常:{}", body.string(), t);
            } else {
                log.error("OpenAI  sse 连接异常 data:{},异常:{}", response, t);
            }
            eventSource.cancel();
        }
    }
    
    目前尚无回复
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   1309 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 23ms · UTC 23:33 · PVG 07:33 · LAX 15:33 · JFK 18:33
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.