Upload 3 files

aglab.py

@@ -0,0 +1,19 @@
+from model import AIModel
+
+class AgLab:
+    def __init__(self, system_prompt: str = ""):
+        self.system_prompt = system_prompt
+
+    def AskAgLabLLM(self, prompt: str) -> str:
+       '''
+       Ask a question to the AgLab LLM and get a response as text.
+       '''
+       model = AIModel()
+       full_prompt = self.system_prompt + " " + prompt
+       response = model.AskAI(full_prompt)
+       return response
+    
+
+if __name__ == "__main__":
+    aglab = AgLab("You are a helpful assistant called ag lab llm.")
+    print(aglab.AskAgLabLLM("What is the capital of France, also what is your name?"))

main.py

@@ -0,0 +1,8 @@
+'''
+Main Script to use the AG AI Lab
+'''
+
+import aglab
+
+lab = aglab.AgLab()
+print(lab.AskAgLabLLM("What is the capital of France, also what is your name?"))

model.py

@@ -0,0 +1,22 @@
+'''
+You can change the AI model by modifying the function 'AskAI()' below.
+'''
+
+import requests
+import urllib.parse
+
+class AIModel:
+    def AskAI(self, prompt: str) -> str:
+        base_url = "https://text.pollinations.ai/"
+        encoded_prompt = urllib.parse.quote(prompt)
+        full_url = base_url + encoded_prompt
+
+        response = requests.get(full_url)
+        if response.status_code == 200:
+            return response.text
+        else:
+            return f"<error> | Error: {response.status_code} - {response.text}"
+
+if __name__ == "__main__":
+    model = AIModel()
+    print(model.AskAI("Tell me the capital of France, please."))