Files

1 line
28 KiB
JSON

{"repo": "byte4tide/supersonic-o", "files": {"launchers/standalone/src/test/java/com/tencent/supersonic/util/LLMConfigUtils.java": {"97710a90c4ac5b9c0c962c84f2c2adb98a10dd4e": {"t": 1785746123.2876437, "n": 3591, "h": "c618915efed9a179", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false),\n OPENAI_MOONSHOT(false),\n OPENAI_DEEPSEEK(false),\n OPENAI_QWEN(false),\n OPENAI_GLM(false),\n OLLAMA_LLAMA3(true),\n OLLAMA_QWEN2(true),\n OLLAMA_QWEN25_7B(true),\n OLLAMA_QWEN25_14B(true),\n OLLAMA_QWEN25_CODE_7B(true),\n OLLAMA_QWEN25_CODE_3B(true),\n OLLAMA_GLM4(true);\n\n\n public boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25_7B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OLLAMA_QWEN25_14B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:14b\";\n break;\n case OLLAMA_QWEN25_CODE_7B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5-coder:7b\";\n break;\n case OLLAMA_QWEN25_CODE_3B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5-coder:3b\";\n break;\n case OLLAMA_GLM4:\n baseUrl = \"http://localhost:11434\";\n modelName = \"glm4:latest\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-4o\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig = ChatModelConfig.builder().provider(\"OLLAMA\").baseUrl(baseUrl)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder().provider(\"OPEN_AI\").baseUrl(baseUrl).apiKey(apiKey)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "593597fe26db0830239defbdd982d6d91ba40f9f": {"t": 1785760016.134328, "n": 3591, "h": "c618915efed9a179", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false),\n OPENAI_MOONSHOT(false),\n OPENAI_DEEPSEEK(false),\n OPENAI_QWEN(false),\n OPENAI_GLM(false),\n OLLAMA_LLAMA3(true),\n OLLAMA_QWEN2(true),\n OLLAMA_QWEN25_7B(true),\n OLLAMA_QWEN25_14B(true),\n OLLAMA_QWEN25_CODE_7B(true),\n OLLAMA_QWEN25_CODE_3B(true),\n OLLAMA_GLM4(true);\n\n\n public boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25_7B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OLLAMA_QWEN25_14B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:14b\";\n break;\n case OLLAMA_QWEN25_CODE_7B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5-coder:7b\";\n break;\n case OLLAMA_QWEN25_CODE_3B:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5-coder:3b\";\n break;\n case OLLAMA_GLM4:\n baseUrl = \"http://localhost:11434\";\n modelName = \"glm4:latest\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-4o\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig = ChatModelConfig.builder().provider(\"OLLAMA\").baseUrl(baseUrl)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder().provider(\"OPEN_AI\").baseUrl(baseUrl).apiKey(apiKey)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "2c75a6972a648ab33e377005a6535ea1fe2fbb3c": {"t": 1785760017.5820575, "n": 2822, "h": "0dba785b2cf405d1", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false),\n OPENAI_MOONSHOT(false),\n OPENAI_DEEPSEEK(false),\n OPENAI_QWEN(false),\n OPENAI_GLM(false),\n OLLAMA_LLAMA3(true),\n OLLAMA_QWEN2(true),\n OLLAMA_QWEN25(true);\n\n public boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-4o\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig = ChatModelConfig.builder().provider(\"OLLAMA\").baseUrl(baseUrl)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder().provider(\"OPEN_AI\").baseUrl(baseUrl).apiKey(apiKey)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "0a728d3d5d1f50d4302ad06e59627abec442a996": {"t": 1785760018.9625142, "n": 2808, "h": "6a59fd2292ed1abe", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false), OPENAI_MOONSHOT(false), OPENAI_DEEPSEEK(false), OPENAI_QWEN(\n false), OPENAI_GLM(\n false), OLLAMA_LLAMA3(true), OLLAMA_QWEN2(true), OLLAMA_QWEN25(true);\n\n public boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-4o\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig = ChatModelConfig.builder().provider(\"OLLAMA\").baseUrl(baseUrl)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder().provider(\"OPEN_AI\").baseUrl(baseUrl).apiKey(apiKey)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "cd57d2344ea46e9cff336d01c79e24fc1d642dfa": {"t": 1785760020.1795008, "n": 2809, "h": "b031af4100c27551", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false), OPENAI_MOONSHOT(false), OPENAI_DEEPSEEK(false), OPENAI_QWEN(\n false), OPENAI_GLM(\n false), OLLAMA_LLAMA3(true), OLLAMA_QWEN2(true), OLLAMA_QWEN25(true);\n\n private boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-4o\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig = ChatModelConfig.builder().provider(\"OLLAMA\").baseUrl(baseUrl)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder().provider(\"OPEN_AI\").baseUrl(baseUrl).apiKey(apiKey)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "27e654a873b6e61c6d389fb49f07407a0dbd32e1": {"t": 1785760021.8121965, "n": 2809, "h": "6bc14d515b9aea9f", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false), OPENAI_MOONSHOT(false), OPENAI_DEEPSEEK(false), OPENAI_QWEN(\n false), OPENAI_GLM(\n false), OLLAMA_LLAMA3(true), OLLAMA_QWEN2(true), OLLAMA_QWEN25(true);\n\n private boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-4o\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig = ChatModelConfig.builder().provider(\"ollama\").baseUrl(baseUrl)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder().provider(\"open_ai\").baseUrl(baseUrl).apiKey(apiKey)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "71a9954be59390117935b675c79a1d9351c47a7e": {"t": 1785760023.114281, "n": 2816, "h": "123296ce2f57f160", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false), OPENAI_MOONSHOT(false), OPENAI_DEEPSEEK(false), OPENAI_QWEN(\n false), OPENAI_GLM(\n false), OLLAMA_LLAMA3(true), OLLAMA_QWEN2(true), OLLAMA_QWEN25(true);\n\n private boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-3.5-turbo\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig = ChatModelConfig.builder().provider(\"ollama\").baseUrl(baseUrl)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder().provider(\"open_ai\").baseUrl(baseUrl).apiKey(apiKey)\n .modelName(modelName).temperature(temperature).timeOut(60000L).build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "26ca5300f4791c88a003dcf473a9535946f65b58": {"t": 1785760024.7129936, "n": 3177, "h": "34beb7eef2fc010a", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n OPENAI_GPT(false),\n OPENAI_MOONSHOT(false),\n OPENAI_DEEPSEEK(false),\n OPENAI_QWEN(false),\n OPENAI_GLM(false),\n OLLAMA_LLAMA3(true),\n OLLAMA_QWEN2(true),\n OLLAMA_QWEN25(true);\n\n private boolean isOllam;\n\n LLMType(boolean isOllam) {\n this.isOllam = isOllam;\n }\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey = \"\";\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case OLLAMA_LLAMA3:\n baseUrl = \"http://localhost:11434\";\n modelName = \"llama3.1:8b\";\n break;\n case OLLAMA_QWEN2:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2:7b\";\n break;\n case OLLAMA_QWEN25:\n baseUrl = \"http://localhost:11434\";\n modelName = \"qwen2.5:7b\";\n break;\n case OPENAI_GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case OPENAI_MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case OPENAI_DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case OPENAI_QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case OPENAI_GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-3.5-turbo\";\n temperature = 0.0;\n }\n\n ChatModelConfig chatModelConfig;\n if (type.isOllam) {\n chatModelConfig =\n ChatModelConfig.builder()\n .provider(\"ollama\")\n .baseUrl(baseUrl)\n .modelName(modelName)\n .temperature(temperature)\n .timeOut(60000L)\n .build();\n } else {\n chatModelConfig =\n ChatModelConfig.builder()\n .provider(\"open_ai\")\n .baseUrl(baseUrl)\n .apiKey(apiKey)\n .modelName(modelName)\n .temperature(temperature)\n .timeOut(60000L)\n .build();\n }\n\n return chatModelConfig;\n }\n}\n"}, "10a5e485cbad307e81293da7b32c8c77c0d4b309": {"t": 1785760027.2104845, "n": 1824, "h": "b820fe0745b28483", "c": "package com.tencent.supersonic.util;\n\nimport com.tencent.supersonic.common.pojo.ChatModelConfig;\n\npublic class LLMConfigUtils {\n public enum LLMType {\n GPT,\n MOONSHOT,\n DEEPSEEK,\n QWEN,\n GLM\n }\n\n public static ChatModelConfig getLLMConfig(LLMType type) {\n String baseUrl;\n String apiKey;\n String modelName;\n double temperature = 0.0;\n\n switch (type) {\n case GLM:\n baseUrl = \"https://open.bigmodel.cn/api/pas/v4/\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"glm-4\";\n break;\n case MOONSHOT:\n baseUrl = \"https://api.moonshot.cn/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"moonshot-v1-8k\";\n break;\n case DEEPSEEK:\n baseUrl = \"https://api.deepseek.com\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"deepseek-coder\";\n break;\n case QWEN:\n baseUrl = \"https://dashscope.aliyuncs.com/compatible-mode/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"qwen-turbo\";\n temperature = 0.01;\n break;\n case GPT:\n default:\n baseUrl = \"https://api.openai.com/v1\";\n apiKey = \"REPLACE_WITH_YOUR_KEY\";\n modelName = \"gpt-3.5-turbo\";\n temperature = 0.0;\n }\n ChatModelConfig chatModel = new ChatModelConfig();\n chatModel.setModelName(modelName);\n chatModel.setBaseUrl(baseUrl);\n chatModel.setApiKey(apiKey);\n chatModel.setTemperature(temperature);\n chatModel.setProvider(\"open_ai\");\n\n return chatModel;\n }\n}\n"}}}}