001/**
002 * Copyright (c) 2025-2026, Michael Yang 杨福海 (fuhai999@gmail.com).
003 * <p>
004 * Licensed under the GNU Lesser General Public License (LGPL) ,Version 3.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 * <p>
008 * http://www.gnu.org/licenses/lgpl-3.0.txt
009 * <p>
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package dev.tinyflow.core.searchengine.impl;
017
018import com.agentsflex.core.chain.Chain;
019import com.agentsflex.core.document.Document;
020import com.agentsflex.core.util.Maps;
021import dev.tinyflow.core.node.SearchEngineNode;
022import dev.tinyflow.core.searchengine.BaseSearchEngine;
023
024import java.util.Collections;
025import java.util.HashMap;
026import java.util.List;
027import java.util.Map;
028
029public class BochaaiSearchEngineImpl extends BaseSearchEngine {
030
031    private static final String DEFAULT_API_URL = "https://api.bochaai.com/v1/ai-search";
032
033    public BochaaiSearchEngineImpl() {
034        setApiUrl(DEFAULT_API_URL);
035    }
036
037
038    @Override
039    public List<Document> search(String keyword, int limit, SearchEngineNode searchEngineNode, Chain chain) {
040
041        Map<String, String> headers = new HashMap<>();
042        headers.put("Authorization", "Bearer " + apiKey);
043        headers.put("Content-Type", "application/json");
044
045        String jsonString = Maps.of("query", keyword)
046                .set("summary", true).
047                set("freshness", "noLimit")
048                .set("count", limit)
049                .set("stream", false)
050                .toJSON();
051
052        String responseString = httpClient.post(apiUrl, headers, jsonString);
053
054
055        return Collections.emptyList();
056    }
057}