chore: Update vehicle destination in calculate_route function

{core → kitt/core}/__init__.py

@@ -10,6 +10,7 @@ from TTS.api import TTS
 
 os.environ["COQUI_TOS_AGREED"] = "1"
 
+tts_pipeline = None
 
 Voice = namedtuple("voice", ["name", "neutral", "angry", "speed"])
 
@@ -181,8 +182,13 @@ def tts(
     return wavs
 
 
-def tts_gradio(tts_pipeline, text, voice, cache):
+def tts_gradio(text, voice, cache):
+    global tts_pipeline
+    if not tts_pipeline:
+        tts_pipeline = load_tts_pipeline()
+
     voice_path = voice_from_text(voice)
+    speed = speed_from_text(voice)
     (gpt_cond_latent, speaker_embedding) = compute_speaker_embedding(
         voice_path, tts_pipeline.synthesizer.tts_config, tts_pipeline, cache
     )
@@ -193,7 +199,7 @@ def tts_gradio(tts_pipeline, text, voice, cache):
         speaker=None,
         gpt_cond_latent=gpt_cond_latent,
         speaker_embedding=speaker_embedding,
-        speed=1.1,
+        speed=speed,
         # file_path="out.wav",
     )
     return (22050, np.array(out)), dict(text=text, voice=voice)

kitt/skills/__init__.py

@@ -0,0 +1,42 @@
+from datetime import datetime
+import inspect
+
+from .common import execute_function_call, extract_func_args, vehicle as vehicle_obj
+from .weather import get_weather, get_forecast
+from .routing import find_route
+from .poi import search_points_of_interests, search_along_route_w_coordinates
+from .vehicle import vehicle_status
+
+
+
+def date_time_info():
+    """Get the current date and time."""
+    time = getattr(vehicle_obj, "time")
+    date = getattr(vehicle_obj, "date")
+    datetime_obj = datetime.fromisoformat(f"{date}T{time}")
+    human_readable_datetime = datetime_obj.strftime("%I:%M %p %A, %B %d, %Y")
+    return f"It is {human_readable_datetime}."
+
+
+def do_anything_else():
+    """If the user wants to do anything else call this function. If the question doesn't match any of the functions use this one."""
+    return True
+
+
+def format_functions_for_prompt_raven(*functions):
+    """Format functions for use in Prompt Raven.
+    
+    Args:
+    *functions (function): One or more functions to format.
+    """
+    formatted_functions = []
+    for func in functions:
+        signature = f"{func.__name__}{inspect.signature(func)}"
+        docstring = inspect.getdoc(func)
+        formatted_functions.append(
+            f"Function:\n<func_start>{signature}<func_end>\n<docstring_start>\n{docstring}\n<docstring_end>"
+        )
+    return "\n".join(formatted_functions)
+
+
+SKILLS_PROMPT = format_functions_for_prompt_raven(get_weather, get_forecast, find_route, search_points_of_interests)

kitt/skills/common.py

@@ -0,0 +1,62 @@
+import re
+from typing import Union
+
+
+from pydantic_settings import BaseSettings, SettingsConfigDict
+from pydantic import BaseModel
+
+from .. import skills
+
+class Settings(BaseSettings):
+    WEATHER_API_KEY: str
+    TOMTOM_API_KEY: str
+
+    model_config = SettingsConfigDict(env_file=".env")
+
+
+class VehicleStatus(BaseModel):
+    location: str
+    location_coordinates: tuple[float, float] # (latitude, longitude)
+    date: str
+    time: str
+    destination: str
+
+
+def execute_function_call(text: str, dry_run=False) -> str:
+    function_name_match = re.search(r"Call: (\w+)", text)
+    function_name = function_name_match.group(1) if function_name_match else None
+    arguments = eval(f"dict{text.split(function_name)[1].strip()}")
+    function = getattr(skills, function_name) if function_name else None
+
+    if dry_run:
+        print(f"{function_name}(**{arguments})")
+        return "Dry run successful"
+
+    if function:
+        out = function(**arguments)
+    try:
+        if function:
+            out = function(**arguments)
+    except Exception as e:
+        out = str(e)
+    return out
+
+
+def extract_func_args(text: str) -> tuple[str, dict]:
+    function_name_match = re.search(r"Call: (\w+)", text)
+    function_name = function_name_match.group(1) if function_name_match else None
+    if not function_name:
+        raise ValueError("No function name found in text")
+    arguments = eval(f"dict{text.split(function_name)[1].strip()}")
+    return function_name, arguments
+
+
+config = Settings() # type: ignore
+
+vehicle = VehicleStatus(
+    location="Rue Alphonse Weicker, Luxembourg",
+    location_coordinates=(49.505, 6.28111),
+    date="2025-05-06",
+    time="08:00:00",
+    destination="Rue Alphonse Weicker, Luxembourg"
+)