Merge pull request #70 from jahanvir/main

Added Support for Google PaLM API

.env.example

@@ -0,0 +1,3 @@
+OPENAI_API_KEY=
+HUGGINGFACEHUB_API_TOKEN=
+PALM_API_KEY=

docs/docs/examples/palmai-bot.md

@@ -0,0 +1,49 @@
+---
+sidebar_position: 2
+---
+
+# Google PaLM AI bot
+
+This bot makes an API call to PaLMAI and processes the user input. It uses PaLM Chat.
+
+```py
+import os
+from textbase import bot, Message
+from textbase.models import PalmAI
+from typing import List
+
+# Load your PALM API key
+# PALMAI.api_key = ""
+# or from environment variable:
+PalmAI.api_key = os.getenv("PALM_API_KEY")
+
+@bot()
+def on_message(message_history: List[Message], state: dict = None):
+
+    bot_response = PalmAI.generate(
+        message_history=message_history, # Assuming history is the list of user messages
+    )
+
+    response = {
+        "data": {
+            "messages": [
+                {
+                    "data_type": "STRING",
+                    "value": bot_response
+                }
+            ],
+            "state": state
+        },
+        "errors": [
+            {
+                "message": ""
+            }
+        ]
+    }
+
+    return {
+        "status_code": 200,
+        "response": response
+    }
+
+```

examples/palmai-bot/main.py

@@ -0,0 +1,39 @@
+import os
+from textbase import bot, Message
+from textbase.models import PalmAI
+from typing import List
+
+# Load your PALM API key
+# PALMAI.api_key = ""
+# or from environment variable:
+PalmAI.api_key = os.getenv("PALM_API_KEY")
+
+
+@bot()
+def on_message(message_history: List[Message], state: dict = None):
+
+    bot_response = PalmAI.generate(
+        message_history=message_history, # Assuming history is the list of user messages
+    )
+
+    response = {
+        "data": {
+            "messages": [
+                {
+                    "data_type": "STRING",
+                    "value": bot_response
+                }
+            ],
+            "state": state
+        },
+        "errors": [
+            {
+                "message": ""
+            }
+        ]
+    }
+
+    return {
+        "status_code": 200,
+        "response": response
+    }

pyproject.toml

@@ -15,6 +15,7 @@ tabulate = "^0.9.0"
 functions-framework = "^3.4.0"
 yaspin = "^3.0.0"
 pydantic = "^2.3.0"
+google-generativeai = "^0.1.0"
 
 [build-system]
 requires = ["poetry-core"]

textbase/models.py

@@ -1,5 +1,6 @@
 import json
 import openai
+import google.generativeai as palm
 import requests
 import time
 import typing
@@ -143,4 +144,30 @@ def generate(
         data = json.loads(response.text) # parse the JSON data into a dictionary
         message = data['message']
 
-        return message
+        return message
+
+class PalmAI:
+    api_key = None
+
+    @classmethod
+    def generate(
+        cls,
+        message_history: list[Message],
+    ):
+
+        assert cls.api_key is not None, "Palm API key is not set."
+        palm.configure(api_key=cls.api_key)
+
+        filtered_messages = []
+
+        for message in message_history:
+            #list of all the contents inside a single message
+            contents = extract_content_values(message)
+            if contents:
+                filtered_messages.extend(contents)
+
+        #send request to Google Palm chat API 
+        response = palm.chat(messages=filtered_messages)
+
+        print(response)
+        return response.last