Upload 3 files

aglab.py

@@ -4,16 +4,42 @@ class AgLab:
     def __init__(self, system_prompt: str = ""):
         self.system_prompt = system_prompt
 
+    def __raw_ask(self, prompt: str) -> str:
+        '''
+        Internal method: Ask a question to the AgLab LLM and get a response as text.
+        '''
+        model = AIModel()
+        response = model.AskAI(prompt)
+        return response
+
     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
-    
+        '''
+        Ask a question to the AgLab LLM and get a response as text.
+        '''
+        full_prompt = self.system_prompt + " " + prompt
+        response = self.__raw_ask(full_prompt)
+        return response
+
+    def AskAgLabLLMWithContext(self, prompt: str, context: str) -> str:
+        '''
+        Ask a question to the AgLab LLM with context and get a response as text.
+        '''
+        full_prompt = (
+            self.system_prompt +
+            " Context of the chat so far: " + context +
+            " User said: " + prompt
+        )
+        response = self.__raw_ask(full_prompt)
+        return response
+
+    def SummarizeText(self, text: str) -> str:
+        '''
+        Summarize a text using the AgLab LLM and get a response as text.
+        '''
+        full_prompt = " Summarize the following text: " + text + " send only the summary of the previous text, do not reply, send only the summary."
+        response = self.__raw_ask(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?"))
+    print(aglab.AskAgLabLLM("What is the capital of France, also what is your name?"))

main.py

@@ -4,5 +4,5 @@ Main Script to use the AG AI Lab
 
 import aglab
 
-lab = aglab.AgLab()
+lab = aglab.AgLab("Your name is ag lab llm. You are a helpful assistant.")
 print(lab.AskAgLabLLM("What is the capital of France, also what is your name?"))