Auto PR from main to live

app/main.py

@@ -58,7 +58,7 @@ async def index(request: Request):
 # Add an get endpoint simple return the evrsion of the app
 @app.get("/version")
 async def version():
-    return {"version": "0.0.2"}
+    return {"version": "0.0.3"}
 
 
 if __name__ == "__main__":

cookbook/ai_assistant_custome_tools.py

@@ -0,0 +1,123 @@
+
+import os
+from typing import Optional, List
+from phi.llm.openai.like import OpenAILike
+from phi.assistant import Assistant
+from phi.knowledge.json import JSONKnowledgeBase
+from phi.vectordb.pgvector import PgVector2
+from phi.storage.assistant.postgres import PgAssistantStorage
+from phi.tools import Toolkit
+from phi.tools.email import EmailTools
+from phi.utils.log import logger
+from phi.tools.email import EmailTools
+from resources import vector_db
+from rich.prompt import Prompt
+from dotenv import load_dotenv
+load_dotenv()
+
+# To run this example, first make sure to follow the instructions below:
+# 1. Install the phidata: pip install phidata
+# 2. Run the following command to start a docker, with pgvector db running: phi start resources.py 
+
+class CinemaTools(Toolkit):
+    def __init__(
+        self,
+        email_tools: Optional["EmailTools"] = None,
+    ):
+        super().__init__(name="cinema_tools")
+        self.email_tools = email_tools
+        self.register(self.book_cinema_ticket)
+
+    def book_cinema_ticket(self, movie_name: str, date: str, time: str, user_email: Optional[str] = None) -> str:
+        """Books a cinema ticket for the given movie, date, and time, and sends an email to the user.
+
+        :param movie_name: The name of the movie.
+        :param date: The date of the movie (e.g., "2023-06-15").
+        :param time: The time of the movie (e.g., "19:30").
+        :param user_email: The email address of the user.
+        :return: "success" if the ticket was booked and email sent successfully, "error: [error message]" otherwise.
+        """
+        # Simulate booking the ticket
+        ticket_number = self._generate_ticket_number()
+        logger.info(f"Booking ticket for {movie_name} on {date} at {time}")
+
+        # Prepare the email subject and body
+        subject = f"Your ticket for {movie_name}"
+        body = f"Dear user,\n\nYour ticket for {movie_name} on {date} at {time} has been booked.\n\n" \
+               f"Your ticket number is: {ticket_number}\n\nEnjoy the movie!\n\nBest regards,\nThe Cinema Team"
+
+        # Send the email using the EmailTools
+        if not self.email_tools:
+            return "error: No email tools provided"
+        self.email_tools.receiver_email = user_email
+        result = self.email_tools.email_user(subject, body)
+
+        if result.startswith("error"):
+            logger.error(f"Error booking ticket: {result}")
+            return result
+        return "success"
+
+    def _generate_ticket_number(self) -> str:
+        """Generates a dummy ticket number."""
+        import random
+        import string
+        return "".join(random.choices(string.ascii_uppercase + string.digits, k=10))
+
+kb = JSONKnowledgeBase(
+    path="cinemax.json",
+    vector_db=PgVector2(collection="cinemax", db_url=vector_db.get_db_connection_local()),
+)
+storage = PgAssistantStorage(
+    table_name="cinemax_assistant_storage",
+    db_url=vector_db.get_db_connection_local(),
+)
+
+my_groq = OpenAILike(
+        model="mixtral-8x7b-32768",
+        api_key=os.environ["GROQ_API_KEY"],
+        base_url="http://localhost:8000/proxy/groq/v1"
+        # base_url="http://groqcall.ai/proxy/groq/v1"
+    )
+
+
+def cinemax_assistant(new: bool = False, user: str = "user"):
+    run_id: Optional[str] = None
+    # new = False
+    # new = True
+    user_id = user
+
+    if not new:
+        existing_run_ids: List[str] = storage.get_all_run_ids(user_id)
+        if len(existing_run_ids) > 0:
+            run_id = existing_run_ids[0]
+
+    assistant = Assistant(
+        run_id=run_id,
+        user_id="test_user",
+        llm=my_groq,
+        knowledge_base=kb,
+        storage=storage,
+        use_tools=True,
+        add_chat_history_to_messages=True,
+        tools=[CinemaTools(EmailTools("YOUR_EMAIL_ADDRESS", "SENDER_NAME", "SENDER_EMAIL", os.environ['email_pass_key'] ))], show_tool_calls=True, markdown=True
+        # add_references_to_prompt=True,
+    )
+    assistant.knowledge_base.load(recreate=False)
+
+    if run_id is None:
+        run_id = assistant.run_id
+        print(f"Started Run: {run_id}\n")
+    else:
+        print(f"Continuing Run: {run_id}\n")
+
+    while True:
+        message = Prompt.ask(f"[bold] :sunglasses: {user} [/bold]")
+        if message in ("exit", "bye"):
+            break
+        assistant.print_response(message, markdown=True, stream=False)
+        # assistant.run(message, markdown=True, stream=False)
+
+if __name__ == "__main__":
+    cinemax_assistant(user="Tom")
+
+

cookbook/resources.py

@@ -0,0 +1,14 @@
+from phi.docker.app.postgres import PgVectorDb
+from phi.docker.resources import DockerResources
+
+# -*- PgVector2 running on port 5432:5432
+vector_db = PgVectorDb(
+    name="knowledge-db",
+    pg_user="ai",
+    pg_password="ai",
+    pg_database="ai",
+    host_port=5532,
+)
+
+# -*- DockerResources
+dev_docker_resources = DockerResources(apps=[vector_db])