Loop44/domotic-agent
1
0
Fork 0
forked from enne2/qbit-agent
Code Issues Pull Requests Projects Releases Wiki Activity

fix: update language model initialization and refine system prompts for better user interaction

Browse Source
This commit is contained in:
Matteo Benedetto
2025-04-18 23:43:00 +02:00
parent d960445d38
commit 735cc4dc44
2 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/search.py
+14 -9
View File
@@ -9,12 +9,14 @@ import time
class MediaInfoSearchTool(BaseTool):
name: str = "duckduckgo_search"
description: str = '''Useful for searching the web using DuckDuckGo for information about \
movies and TV shows, actors and directors.'''
name: str = "media_info_search"
description: str = '''Useful for searching on trustworthy sites for movie information, reviews, and similar content to a given title or plot.\
Use when the user asks for a movie or TV series information, reviews, plots, casts members, or similar content.\
Input should be a search query, and the tool will return relevant results.'''
# Class variable to track previous queries and sites
movie_sites: List[str] = ["imdb.com", "rottentomatoes.com", "metacritic.com", "themoviedb.org", "filmaffinity.com"]
movie_sites: List[str] = ["imdb.com", "rottentomatoes.com", "metacritic.com", "themoviedb.org", "filmaffinity.com", "letterboxd.com",
"boxofficemojo.com", "fandango.com", "movieinsider.com", "comingsoon.net", "wikipedia.org", "tvguide.com"]
def _run(self, query: str, run_manager: Optional[CallbackManagerForToolRun] = None) -> str:
"""Perform a DuckDuckGo search with site rotation queries."""
@@ -22,13 +24,14 @@ class MediaInfoSearchTool(BaseTool):
search_tool = DuckDuckGoSearchRun()
result = ""
# Randomly select 3 sites from the movie_sites list
selected_sites = random.sample(self.movie_sites, 2)
selected_sites = random.sample(self.movie_sites, 2
)
for movie_site in selected_sites:
result += f"Searching for '{query}' on {movie_site}...\n"
try:
# Perform the search using DuckDuckGo
result += search_tool.run(f"{query} site:{movie_site}")
result += search_tool.run(f"{query} {movie_site}")
except Exception as e:
result += f"Error searching on {movie_site}: {str(e)}\n"
time.sleep(1) # Sleep for 1 second to avoid hitting the API too fast
@@ -43,12 +46,13 @@ class MediaInfoSearchTool(BaseTool):
class MoviesAdviceSearchTool(BaseTool):
name: str = "movies_advice_search"
description: str = '''Useful for searching the web using DuckDuckGo for movie recommendations and similar content to a given title or plot.
description: str = '''Useful for searching the web using DuckDuckGo for movie recommendations to a given title or plot.\
Use ONLYif user appears to be looking for a new movie or TV series to watch.\
prefer searching on trustworthy sites.
Input should be a search query, and the tool will return relevant results.'''
# Class variable to track recommendation sites
recommendation_sites: List[str] = ["reddit.com/r/moviesuggestions", "tastedive.com", "letterboxd.com", "movielens.org", "flickmetrix.com", "justwatch.com"]
recommendation_sites: List[str] = ["reddit.com/r/moviesuggestions", "tastedive.com", "letterboxd.com", "movielens.org", "justwatch.com"]
def _run(self, query: str, run_manager: Optional[CallbackManagerForToolRun] = None) -> str:
"""Perform a DuckDuckGo search with site rotation queries."""
@@ -62,7 +66,8 @@ class MoviesAdviceSearchTool(BaseTool):
result += f"Searching for '{query}' on {rec_site}...\n"
try:
# Perform the search using DuckDuckGo
result += search_tool.run(f"{query} site:{rec_site}")
result += search_tool.run(f"{query} {rec_site}")
except Exception as e:
result += f"Error searching on {rec_site}: {str(e)}\n"
time.sleep(1) # Sleep for 1 second to avoid hitting the API too fast