Patch 🩹

Author: bayeggex

examples/Datatest/Datatest.ino

@@ -9,12 +9,7 @@ void setup() {
     chatbot.setKey("YOUR_CHATGPT_API_KEY", "chatgpt");
 
     //Begin WiFi connection
-    WiFi.begin("SSID", "PASSWORD");
-      while (WiFi.status() != WL_CONNECTED) {
-         delay(1000);
-         Serial.println("Connecting to WiFi...");
-    }
-    Serial.println("Connected to WiFi");
+    chatbot.connectWiFi("YOUR_SSID", "YOUR_PASSWORD");
 
     // Select AI and optionally specify version
     chatbot.selectAI("chatgpt", "gpt-3.5-turbo");

library.properties

@@ -1,10 +1,10 @@
 name=AIChatBot
-version=0.1.0
+version=1.1.0
 author=bay_Eggex <https://github.com/bayeggex>
 maintainer=bay_Eggex
 sentence=AI Chat Library for Arduino 
 paragraph=Arduino library for integrating AI chat like OpenAIs ChatGPT, Hugging Face Transformers, and more.
 category=Communication
 url=https://github.com/bayeggex/Arduino-AI-Chat-Library
 architectures=*
-includes=AIChatbot.h
+includes=AIChatbot.h

src/AIChatbot.cpp

@@ -1,15 +1,19 @@
 #include "AIChatbot.h"
 
-AIChatbot::AIChatbot() : chatGPTApiKey(""), huggingFaceApiKey(""), selectedAI(""), selectedAIVersion("gpt-3.5-turbo") {}
+AIChatbot::AIChatbot()
+    : chatGPTApiKey(""), huggingFaceApiKey(""), selectedAI(""), selectedAIVersion("gpt-3.5-turbo") {}
 
 void AIChatbot::begin(long baudRate) {
     Serial.begin(baudRate);
-    // WiFi.begin("SSID", "PASSWORD");
-    // while (WiFi.status() != WL_CONNECTED) {
-    //     delay(1000);
-    //     Serial.println("Connecting to WiFi...");
-    // }
-    // Serial.println("Connected to WiFi");
+}
+
+void AIChatbot::connectWiFi(const char* ssid, const char* password) {
+    WiFi.begin(ssid, password);
+    while (WiFi.status() != WL_CONNECTED) {
+        delay(1000);
+        Serial.println("Connecting to WiFi...");
+    }
+    Serial.println("Connected to WiFi");
 }
 
 void AIChatbot::update() {
@@ -54,13 +58,6 @@ String AIChatbot::sendToChatGPT(const String& message) {
         return "ChatGPT API key not set.";
     }
 
-    if(selectedAIVersion.isEmpty()) {
-        return "Check the Spelling or Write the name of the AI Model Correctly, if is not take a look catalog ""https://platform.openai.com/docs/models/overview";
-    }
-    if(selectedAIVersion == "DALL" && "DALL-E" && "Whisper"){
-        return "Its not supported at this point, try to use gpt-4-turbo, gpt-3.5-turbo or other GPT modules, also take a look catalog ""https://platform.openai.com/docs/models/overview"" for more information";
-    }
-
     String url = "https://api.openai.com/v1/chat/completions";
     String payload = "{\"model\": \"" + selectedAIVersion + "\", \"messages\": [{\"role\": \"user\", \"content\": \"" + message + "\"}]}";
 
@@ -72,16 +69,6 @@ String AIChatbot::sendToHuggingFace(const String& message) {
         return "Hugging Face API key not set.";
     }
 
-    if(selectedAIVersion.isEmpty()) {
-        return "Check the Spelling or Write the name of the AI Model Correctly, if is not take a look catalog ""https://huggingface.co/models";
-    }
-
-    if (huggingFaceApiKey.isEmpty() && selectedAIVersion.isEmpty())
-    {
-        return "Hugging Face API key and AI Model not set. please look up the doc ""https://github.com/bayeggex/Arduino-AI-Chat-Library/blob/main/Docs.md""";
-    }
-    
-
     String url = "https://api-inference.huggingface.co/models/" + selectedAIVersion;
     String payload = "{\"inputs\": \"" + message + "\"}";
 

src/AIChatbot.h

@@ -2,13 +2,20 @@
 #define AI_CHATBOT_H
 
 #include <Arduino.h>
+
+#ifdef ESP32
 #include <WiFi.h>
 #include <HTTPClient.h>
+#elif defined(ESP8266)
+#include <ESP8266WiFi.h>
+#include <ESP8266HTTPClient.h>
+#endif
 
 class AIChatbot {
 public:
     AIChatbot();
     void begin(long baudRate);
+    void connectWiFi(const char* ssid, const char* password);
     void update();
     void setKey(const String& key, const String& aiName);
     void selectAI(const String& aiName, const String& aiVersion = "gpt-3.5-turbo");
@@ -24,4 +31,4 @@ class AIChatbot {
     String makeHttpRequest(const String& url, const String& payload, const String& apiKey);
 };
 
-#endif
+#endif // AI_CHATBOT_H