git.basschouten.com Git - openhab-addons.git/blob

   1 /**
   2  * Copyright (c) 2010-2023 Contributors to the openHAB project
   3  *
   4  * See the NOTICE file(s) distributed with this work for additional
   5  * information.
   6  *
   7  * This program and the accompanying materials are made available under the
   8  * terms of the Eclipse Public License 2.0 which is available at
   9  * http://www.eclipse.org/legal/epl-2.0
  10  *
  11  * SPDX-License-Identifier: EPL-2.0
  12  */
  13 package org.openhab.binding.chatgpt.internal;
  14
  15 import static org.openhab.binding.chatgpt.internal.ChatGPTBindingConstants.*;
  16
  17 import java.util.ArrayList;
  18 import java.util.Collection;
  19 import java.util.List;
  20 import java.util.concurrent.ExecutionException;
  21 import java.util.concurrent.TimeoutException;
  22
  23 import org.eclipse.jdt.annotation.NonNullByDefault;
  24 import org.eclipse.jdt.annotation.Nullable;
  25 import org.eclipse.jetty.client.HttpClient;
  26 import org.eclipse.jetty.client.api.ContentResponse;
  27 import org.eclipse.jetty.client.api.Request;
  28 import org.eclipse.jetty.client.util.StringContentProvider;
  29 import org.eclipse.jetty.http.HttpMethod;
  30 import org.eclipse.jetty.http.HttpStatus;
  31 import org.openhab.binding.chatgpt.internal.dto.ChatResponse;
  32 import org.openhab.core.io.net.http.HttpClientFactory;
  33 import org.openhab.core.library.types.StringType;
  34 import org.openhab.core.thing.Channel;
  35 import org.openhab.core.thing.ChannelUID;
  36 import org.openhab.core.thing.Thing;
  37 import org.openhab.core.thing.ThingStatus;
  38 import org.openhab.core.thing.ThingStatusDetail;
  39 import org.openhab.core.thing.binding.BaseThingHandler;
  40 import org.openhab.core.thing.binding.ThingHandlerService;
  41 import org.openhab.core.types.Command;
  42 import org.openhab.core.types.RefreshType;
  43 import org.slf4j.Logger;
  44 import org.slf4j.LoggerFactory;
  45
  46 import com.google.gson.Gson;
  47 import com.google.gson.JsonArray;
  48 import com.google.gson.JsonElement;
  49 import com.google.gson.JsonObject;
  50
  51 /**
  52  * The {@link ChatGPTHandler} is responsible for handling commands, which are
  53  * sent to one of the channels.
  54  *
  55  * @author Kai Kreuzer - Initial contribution
  56  */
  57 @NonNullByDefault
  58 public class ChatGPTHandler extends BaseThingHandler {
  59
  60     private final Logger logger = LoggerFactory.getLogger(ChatGPTHandler.class);
  61
  62     private HttpClient httpClient;
  63     private Gson gson = new Gson();
  64
  65     private String apiKey = "";
  66     private String lastPrompt = "";
  67
  68     private List<String> models = List.of();
  69
  70     public ChatGPTHandler(Thing thing, HttpClientFactory httpClientFactory) {
  71         super(thing);
  72         this.httpClient = httpClientFactory.getCommonHttpClient();
  73     }
  74
  75     @Override
  76     public void handleCommand(ChannelUID channelUID, Command command) {
  77         if (command instanceof RefreshType && !"".equals(lastPrompt)) {
  78             String response = sendPrompt(channelUID, lastPrompt);
  79             processChatResponse(channelUID, response);
  80         }
  81
  82         if (command instanceof StringType stringCommand) {
  83             lastPrompt = stringCommand.toFullString();
  84             String response = sendPrompt(channelUID, lastPrompt);
  85             processChatResponse(channelUID, response);
  86         }
  87     }
  88
  89     private void processChatResponse(ChannelUID channelUID, @Nullable String response) {
  90         if (response != null) {
  91             ChatResponse chatResponse = gson.fromJson(response, ChatResponse.class);
  92             if (chatResponse != null) {
  93                 String msg = chatResponse.getChoices().get(0).getMessage().getContent();
  94                 updateState(channelUID, new StringType(msg));
  95             } else {
  96                 logger.warn("Didn't receive any response from ChatGPT - this is unexpected.");
  97             }
  98         }
  99     }
 100
 101     private @Nullable String sendPrompt(ChannelUID channelUID, String prompt) {
 102         Channel channel = getThing().getChannel(channelUID);
 103         if (channel == null) {
 104             logger.error("Channel with UID '{}' cannot be found on Thing '{}'.", channelUID, getThing().getUID());
 105             return null;
 106         }
 107         ChatGPTChannelConfiguration channelConfig = channel.getConfiguration().as(ChatGPTChannelConfiguration.class);
 108
 109         JsonObject root = new JsonObject();
 110         root.addProperty("temperature", channelConfig.temperature);
 111         root.addProperty("model", channelConfig.model);
 112         root.addProperty("max_tokens", channelConfig.maxTokens);
 113
 114         JsonObject systemMessage = new JsonObject();
 115         systemMessage.addProperty("role", "system");
 116         systemMessage.addProperty("content", channelConfig.systemMessage);
 117         JsonObject userMessage = new JsonObject();
 118         userMessage.addProperty("role", "user");
 119         userMessage.addProperty("content", prompt);
 120         JsonArray messages = new JsonArray(2);
 121         messages.add(systemMessage);
 122         messages.add(userMessage);
 123         root.add("messages", messages);
 124
 125         Request request = httpClient.newRequest(OPENAI_API_URL).method(HttpMethod.POST)
 126                 .header("Content-Type", "application/json").header("Authorization", "Bearer " + apiKey)
 127                 .content(new StringContentProvider(gson.toJson(root)));
 128         try {
 129             ContentResponse response = request.send();
 130             updateStatus(ThingStatus.ONLINE);
 131             if (response.getStatus() == HttpStatus.OK_200) {
 132                 return response.getContentAsString();
 133             } else {
 134                 logger.error("ChatGPT request resulted in HTTP {} with message: {}", response.getStatus(),
 135                         response.getReason());
 136                 return null;
 137             }
 138         } catch (InterruptedException | TimeoutException | ExecutionException e) {
 139             updateStatus(ThingStatus.OFFLINE, ThingStatusDetail.COMMUNICATION_ERROR,
 140                     "Could not connect to OpenAI API: " + e.getMessage());
 141             logger.debug("Request to OpenAI failed: {}", e.getMessage(), e);
 142             return null;
 143         }
 144     }
 145
 146     @Override
 147     public void initialize() {
 148         ChatGPTConfiguration config = getConfigAs(ChatGPTConfiguration.class);
 149
 150         String apiKey = config.apiKey;
 151
 152         if (apiKey.isBlank()) {
 153             updateStatus(ThingStatus.OFFLINE, ThingStatusDetail.CONFIGURATION_ERROR,
 154                     "@text/offline.configuration-error");
 155             return;
 156         }
 157
 158         this.apiKey = apiKey;
 159         updateStatus(ThingStatus.UNKNOWN);
 160
 161         scheduler.execute(() -> {
 162             try {
 163                 Request request = httpClient.newRequest(OPENAI_MODELS_URL).method(HttpMethod.GET)
 164                         .header("Authorization", "Bearer " + apiKey);
 165                 ContentResponse response = request.send();
 166                 if (response.getStatus() == 200) {
 167                     updateStatus(ThingStatus.ONLINE);
 168                     JsonObject jsonObject = gson.fromJson(response.getContentAsString(), JsonObject.class);
 169                     if (jsonObject != null) {
 170                         JsonArray data = jsonObject.getAsJsonArray("data");
 171
 172                         List<String> modelIds = new ArrayList<>();
 173                         for (JsonElement element : data) {
 174                             JsonObject model = element.getAsJsonObject();
 175                             String id = model.get("id").getAsString();
 176                             modelIds.add(id);
 177                         }
 178                         this.models = List.copyOf(modelIds);
 179                     } else {
 180                         logger.warn("Did not receive a valid JSON response from the models endpoint.");
 181                     }
 182                 } else {
 183                     updateStatus(ThingStatus.OFFLINE, ThingStatusDetail.COMMUNICATION_ERROR,
 184                             "@text/offline.communication-error");
 185                 }
 186             } catch (InterruptedException | ExecutionException | TimeoutException e) {
 187                 updateStatus(ThingStatus.OFFLINE, ThingStatusDetail.COMMUNICATION_ERROR, e.getMessage());
 188             }
 189         });
 190     }
 191
 192     List<String> getModels() {
 193         return models;
 194     }
 195
 196     @Override
 197     public Collection<Class<? extends ThingHandlerService>> getServices() {
 198         return List.of(ChatGPTModelOptionProvider.class);
 199     }
 200 }