Spaces:
Configuration error
Configuration error
File size: 32,072 Bytes
81cc195 949ee6f ffa5780 c0d2b2f 81cc195 cac1202 c0d2b2f 81cc195 e4bc671 81cc195 a8a67ee 81cc195 9da53da bc5865c 81cc195 8775849 eb8cfb8 81cc195 3935bca 81cc195 cac1202 8b1644b 8012613 25b9c20 c0d2b2f 81cc195 c0d2b2f 5fa5125 46264e6 c0d2b2f 4d5fb54 9da53da 55ae385 c0d2b2f 55ae385 c0d2b2f 5aecaaa 627d948 c0d2b2f 3e14681 c0d2b2f 627d948 c0d2b2f 0419c04 0b15a3d c02a77b 1a3b775 4c11be0 c0d2b2f c02a77b c0d2b2f c9df553 c0d2b2f 36bdc91 8d75eb4 c0d2b2f 197acc2 d52ba4d 8d75eb4 98e38b0 9da53da 98e38b0 c0d2b2f 9da53da c0d2b2f 9da53da 98e38b0 c0d2b2f 9da53da c0d2b2f 9da53da 9bba232 c0d2b2f 9da53da 9bba232 9da53da c0d2b2f 9da53da c0d2b2f 9da53da 9bba232 9da53da 9bba232 c0d2b2f 9da53da 98e38b0 9da53da 98e38b0 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da cac1202 9da53da c0d2b2f cac1202 c0d2b2f cac1202 c0d2b2f cac1202 c0d2b2f cac1202 c0d2b2f cac1202 a8a67ee cac1202 c0d2b2f cac1202 40c30ca c0d2b2f cac1202 c0d2b2f cac1202 c0d2b2f 9da53da c0d2b2f bd4ad64 9da53da c0d2b2f 9da53da c5a7c47 5f75978 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da 36bdc91 9da53da 36bdc91 9da53da c0d2b2f 9da53da 949ee6f c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c9496f6 98e38b0 949ee6f 8012613 9da53da 98e38b0 9da53da 98e38b0 9da53da 8012613 81a79f6 9da53da c0d2b2f 9da53da 36bdc91 9da53da 3935bca cac1202 c0d2b2f 36bdc91 c0d2b2f 36bdc91 84eb4f4 0ee7b8a 84eb4f4 c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da 627d948 817a872 627d948 03b1b87 627d948 c0d2b2f 627d948 817a872 627d948 8da3d65 c0d2b2f 075a9b5 8da3d65 627d948 8da3d65 817a872 c0d2b2f 817a872 c0d2b2f 817a872 c0d2b2f 817a872 81cc195 c0d2b2f 81cc195 817a872 c0d2b2f 817a872 c0d2b2f 817a872 81cc195 c0d2b2f 817a872 c0d2b2f 817a872 81cc195 c0d2b2f 81cc195 c0d2b2f 81cc195 c0d2b2f 81cc195 c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9c24754 c0d2b2f 9c24754 c0d2b2f 9c24754 c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 9da53da c0d2b2f 81cc195 9da53da 81cc195 c0d2b2f 9da53da 81cc195 3e72ec9 81cc195 9da53da 3e72ec9 81cc195 c0d2b2f 81cc195 9da53da c0d2b2f 81cc195 9da53da c0d2b2f 81cc195 c0d2b2f 81cc195 c0d2b2f 81cc195 c0d2b2f 81cc195 cb81c91 c0d2b2f cb81c91 81cc195 9da53da c0d2b2f 9da53da fad1a37 08df1af bb1f1b4 c0d2b2f fad1a37 b353147 fad1a37 c0d2b2f 900b89e c02a77b bb1f1b4 fad1a37 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 |
# Standard library imports
import logging
import os
import re
from typing import Dict, Any, List, Optional, Union
from urllib.parse import urlparse
import torch
import asyncio
# Third-party imports
import requests
from transformers import AutoModelForCausalLM, AutoTokenizer
from custom_models import QwenVL7BCustomLLM, BaaiMultimodalEmbedding
# LlamaIndex core imports
from llama_index.core import VectorStoreIndex, Document, Settings
from llama_index.core.agent.workflow import FunctionAgent, ReActAgent, AgentStream
from llama_index.core.node_parser import UnstructuredElementNodeParser, SentenceSplitter
from llama_index.core.postprocessor import SentenceTransformerRerank
from llama_index.core.query_engine import RetrieverQueryEngine
from llama_index.core.retrievers import VectorIndexRetriever
from llama_index.core.tools import FunctionTool
from llama_index.core.workflow import Context
from llama_index.postprocessor.colpali_rerank import ColPaliRerank
from llama_index.core.schema import ImageNode, TextNode
# LlamaIndex specialized imports
from llama_index.embeddings.huggingface import HuggingFaceEmbedding
from llama_index.llms.huggingface import HuggingFaceLLM
from llama_index.readers.assemblyai import AssemblyAIAudioTranscriptReader
from llama_index.readers.json import JSONReader
from llama_index.readers.web import BeautifulSoupWebReader
from llama_index.readers.youtube_transcript import YoutubeTranscriptReader
from llama_index.tools.arxiv import ArxivToolSpec
from llama_index.core.agent.workflow import AgentWorkflow
from llama_index.llms.vllm import Vllm
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
# Import all required official LlamaIndex Readers
from llama_index.readers.file import (
PDFReader,
DocxReader,
CSVReader,
PandasExcelReader,
VideoAudioReader # Adding VideoAudioReader for handling audio/video without API
)
from pydantic import PrivateAttr
# Optional API-based imports (conditionally loaded)
try:
# Gemini (for API mode)
from llama_index.llms.gemini import Gemini
from llama_index.embeddings.gemini import GeminiEmbedding
GEMINI_AVAILABLE = True
except ImportError:
GEMINI_AVAILABLE = False
try:
# LlamaParse for document parsing (API mode)
from llama_cloud_services import LlamaParse
LLAMAPARSE_AVAILABLE = True
except ImportError:
LLAMAPARSE_AVAILABLE = False
import importlib.util
import sys
import weave
weave.init("gaia-llamaindex-agents")
def get_max_memory_config(max_memory_per_gpu):
"""Generate max_memory config for available GPUs"""
if torch.cuda.is_available():
num_gpus = torch.cuda.device_count()
max_memory = {}
for i in range(num_gpus):
max_memory[i] = max_memory_per_gpu
return max_memory
return None
# Initialize models based on API availability
def initialize_models(use_api_mode=False):
"""Initialize LLM, Code LLM, and Embed models based on mode"""
if use_api_mode and GEMINI_AVAILABLE:
# API Mode - Using Google's Gemini models
try:
print("Initializing models in API mode with Gemini...")
google_api_key = os.environ.get("GOOGLE_API_KEY")
if not google_api_key:
print("WARNING: GOOGLE_API_KEY not found. Falling back to non-API mode.")
return initialize_models(use_api_mode=False)
# Main LLM - Gemini 2.0 Flash
proj_llm = Gemini(
model="models/gemini-2.0-flash",
api_key=google_api_key,
max_tokens=16000,
temperature=0.6,
top_p=0.95,
top_k=20
)
# Same model for code since Gemini is good at code
code_llm = proj_llm
# Vertex AI multimodal embedding
embed_model = GeminiEmbedding(
model_name="models/embedding-001",
api_key=google_api_key,
task_type="retrieval_document"
)
return proj_llm, code_llm, embed_model
except Exception as e:
print(f"Error initializing API mode: {e}")
print("Falling back to non-API mode...")
return initialize_models(use_api_mode=False)
else:
# Non-API Mode - Using HuggingFace models
print("Initializing models in non-API mode with local models...")
try :
proj_llm = QwenVL7BCustomLLM()
embed_model = MultimodalCLIPEmbedding()
embed_model.max_seq_length = 1024
# Code LLM
code_llm = HuggingFaceLLM(
model_name="Qwen/Qwen2.5-Coder-3B-Instruct",
tokenizer_name="Qwen/Qwen2.5-Coder-3B-Instruct",
device_map="auto",
model_kwargs={"torch_dtype": "auto"},
generate_kwargs={"do_sample": False}
)
return proj_llm, code_llm, embed_model
except Exception as e:
print(f"Error initializing models: {e}")
raise
# Setup logging
logging.basicConfig(level=logging.INFO)
logging.getLogger("llama_index.core.agent").setLevel(logging.DEBUG)
logging.getLogger("llama_index.llms").setLevel(logging.DEBUG)
# Use environment variable to determine API mode
USE_API_MODE = os.environ.get("USE_API_MODE", "false").lower() == "true"
# Initialize models based on API mode setting
proj_llm, code_llm, embed_model = initialize_models(use_api_mode=USE_API_MODE)
# Set global settings
Settings.llm = proj_llm
Settings.embed_model = embed_model
def read_and_parse_content(input_path: str) -> List[Document]:
"""
Reads and parses content from a local file path into Document objects.
URL handling has been moved to search_and_extract_top_url.
"""
# Check if API mode and LlamaParse is available for enhanced document parsing
if USE_API_MODE and LLAMAPARSE_AVAILABLE:
try:
llamacloud_api_key = os.environ.get("LLAMA_CLOUD_API_KEY")
if llamacloud_api_key:
# Use LlamaParse for enhanced document parsing
print(f"Using LlamaParse to extract content from {input_path}")
parser = LlamaParse(api_key=llamacloud_api_key)
return parser.load_data(input_path)
except Exception as e:
print(f"Error using LlamaParse: {e}")
print("Falling back to standard document parsing...")
# Standard document parsing (fallback)
if not os.path.exists(input_path):
return [Document(text=f"Error: File not found at {input_path}")]
file_extension = os.path.splitext(input_path)[1].lower()
# Readers map
readers_map = {
'.pdf': PDFReader(),
'.docx': DocxReader(),
'.doc': DocxReader(),
'.csv': CSVReader(),
'.json': JSONReader(),
'.xlsx': PandasExcelReader(),
}
# Audio/Video files using the appropriate reader based on mode
if file_extension in ['.mp3', '.mp4', '.wav', '.m4a', '.flac']:
try:
if USE_API_MODE:
# Use AssemblyAI with API mode
loader = AssemblyAIAudioTranscriptReader(file_path=input_path)
documents = loader.load_data()
else:
# Use VideoAudioReader without API
loader = VideoAudioReader()
documents = loader.load_data(input_path)
return documents
except Exception as e:
return [Document(text=f"Error transcribing audio: {e}")]
# Handle image files
if file_extension in ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp']:
try:
with open(input_path, 'rb') as f:
image_data = f.read()
return [Document(
text=f"IMAGE_CONTENT_BINARY",
metadata={
"source": input_path,
"type": "image",
"path": input_path,
"image_data": image_data
}
)]
except Exception as e:
return [Document(text=f"Error reading image: {e}")]
# Use appropriate reader for supported file types
if file_extension in readers_map:
loader = readers_map[file_extension]
documents = loader.load_data(file=input_path)
else:
# Fallback for text files
try:
with open(input_path, 'r', encoding='utf-8') as f:
content = f.read()
documents = [Document(text=content, metadata={"source": input_path})]
except Exception as e:
return [Document(text=f"Error reading file as plain text: {e}")]
# Add source metadata
for doc in documents:
doc.metadata["source"] = input_path
return documents
class DynamicQueryEngineManager:
"""Single unified manager for all RAG operations - replaces the entire static approach."""
def __init__(self, initial_documents: List[str] = None):
self.documents = []
self.query_engine_tool = None
# Load initial documents if provided
if initial_documents:
self._load_initial_documents(initial_documents)
self._create_rag_tool()
def _load_initial_documents(self, document_paths: List[str]):
"""Load initial documents using read_and_parse_content."""
for path in document_paths:
docs = read_and_parse_content(path)
self.documents.extend(docs)
print(f"Loaded {len(self.documents)} initial documents")
def _create_rag_tool(self):
"""Create RAG tool using multimodal-aware parsing."""
documents = self.documents if self.documents else [
Document(text="No documents loaded yet. Use web search to add content.")
]
# Separate text and image documents for proper processing
text_documents = []
image_documents = []
for doc in documents:
doc_type = doc.metadata.get("type", "")
source = doc.metadata.get("source", "").lower()
file_type = doc.metadata.get("file_type", "")
# Identify image documents
if (doc_type in ["image", "web_image"] or
file_type in ['jpg', 'png', 'jpeg', 'gif', 'bmp', 'webp'] or
any(ext in source for ext in ['.jpg', '.png', '.jpeg', '.gif', '.bmp', '.webp'])):
image_documents.append(doc)
else:
text_documents.append(doc)
# Use UnstructuredElementNodeParser for text content with multimodal awareness
element_parser = UnstructuredElementNodeParser()
splitter = SentenceSplitter(chunk_size=1024, chunk_overlap=200)
nodes = []
# Process text documents with UnstructuredElementNodeParser
if text_documents:
initial_nodes = element_parser.get_nodes_from_documents(text_documents)
final_nodes = splitter.get_nodes_from_documents(initial_nodes)
nodes.extend(final_nodes)
# Process image documents as ImageNodes
if image_documents:
for img_doc in image_documents:
try:
image_node = ImageNode(
text=img_doc.text or f"Image content from {img_doc.metadata.get('source', 'unknown')}",
metadata=img_doc.metadata,
image_path=img_doc.metadata.get("path"),
image=img_doc.metadata.get("image_data")
)
nodes.append(image_node)
except Exception as e:
print(f"Error creating ImageNode: {e}")
# Fallback to regular TextNode for images
text_node = TextNode(
text=img_doc.text or f"Image content from {img_doc.metadata.get('source', 'unknown')}",
metadata=img_doc.metadata
)
nodes.append(text_node)
index = VectorStoreIndex(nodes)
class HybridReranker:
def __init__(self):
self.text_reranker = SentenceTransformerRerank(
model="cross-encoder/ms-marco-MiniLM-L-2-v2",
top_n=3,
device = "cpu"
)
self.visual_reranker = ColPaliRerank(
top_n=3,
model="vidore/colpali-v1.2",
keep_retrieval_score=True,
device="cpu"
)
def postprocess_nodes(self, nodes, query_bundle):
# Separate text and visual nodes
text_nodes = []
visual_nodes = []
for node in nodes:
if (hasattr(node, 'image_path') and node.image_path) or (hasattr(node, 'metadata') and node.metadata.get('file_type') in ['jpg', 'png', 'jpeg', 'pdf']) or (hasattr(node, 'metadata') and node.metadata.get('type') in ['image', 'web_image']):
visual_nodes.append(node)
else:
text_nodes.append(node)
# Apply appropriate reranker
reranked_text = []
reranked_visual = []
if text_nodes:
reranked_text = self.text_reranker.postprocess_nodes(text_nodes, query_bundle)
if visual_nodes:
reranked_visual = self.visual_reranker.postprocess_nodes(visual_nodes, query_bundle)
# Interleave results
combined_results = []
max_len = max(len(reranked_text), len(reranked_visual))
for i in range(max_len):
if i < len(reranked_text):
combined_results.append(reranked_text[i])
if i < len(reranked_visual):
combined_results.append(reranked_visual[i])
return combined_results[:5]
hybrid_reranker = HybridReranker()
query_engine = index.as_query_engine(
similarity_top_k=20,
node_postprocessors=[hybrid_reranker],
response_mode="tree_summarize"
)
# Create QueryEngineTool
from llama_index.core.tools import QueryEngineTool
self.query_engine_tool = QueryEngineTool.from_defaults(
query_engine=query_engine,
name="dynamic_hybrid_multimodal_rag_tool",
description=(
"Advanced dynamic knowledge base with hybrid reranking. "
"Uses ColPali for visual content and SentenceTransformer for text content. "
"Automatically updated with web search content."
)
)
def add_documents(self, new_documents: List[Document]):
"""Add documents from web search and recreate tool."""
self.documents.extend(new_documents)
self._create_rag_tool() # Recreate with ALL documents
print(f"Added {len(new_documents)} documents. Total: {len(self.documents)}")
def get_tool(self):
return self.query_engine_tool
# Global instance
dynamic_qe_manager = DynamicQueryEngineManager()
# 1. Create the base DuckDuckGo search tool from the official spec.
# This tool returns text summaries of search results, not just URLs.
base_duckduckgo_tool = DuckDuckGoSearchToolSpec().to_tool_list()[1]
def search_and_extract_content_from_url(query: str) -> List[Document]:
"""
Searches web, gets top URL, and extracts both text content and images.
Returns a list of Document objects containing the extracted content.
"""
# Get URL from search
#search_results = base_duckduckgo_tool(query, max_results=1)
#url_match = re.search(r"https?://\S+", str(search_results))
#if not url_match:
#return [Document(text="No URL could be extracted from the search results.")]
#url = url_match.group(0)[:-2]
url = "https://en.wikipedia.org/wiki/Mercedes_Sosa"
print(url)
documents = []
try:
# Check if it's a YouTube URL
if "youtube" in urlparse(url).netloc or "youtu.be" in urlparse(url).netloc:
loader = YoutubeTranscriptReader()
documents = loader.load_data(youtubelinks=[url])
else:
loader = BeautifulSoupWebReader()
documents = loader.load_data(urls=[url])
for doc in documents:
doc.metadata["source"] = url
doc.metadata["type"] = "web_text"
return documents
except Exception as e:
# Handle any exceptions that occur during content extraction
print(e)
return [Document(text=f"Error extracting content from URL: {str(e)}")]
def enhanced_web_search_and_update(query: str) -> str:
"""
Performs web search, extracts content, and adds it to the dynamic query engine.
"""
# Extract content from web search
documents = search_and_extract_content_from_url(query)
# Add documents to the dynamic query engine
if documents and not any("Error" in doc.text for doc in documents):
dynamic_qe_manager.add_documents(documents)
# Return summary of what was added
text_docs = [doc for doc in documents if doc.metadata.get("type") == "web_text"]
image_docs = [doc for doc in documents if doc.metadata.get("type") == "web_image"]
summary = f"Successfully added web content to knowledge base:\n"
summary += f"- {len(text_docs)} text documents\n"
summary += f"- {len(image_docs)} images\n"
summary += f"Source: {documents[0].metadata.get('source', 'Unknown')}"
return summary
else:
error_msg = documents[0].text if documents else "No content extracted"
return f"Failed to extract web content: {error_msg}"
# Create the enhanced web search tool
enhanced_web_search_tool = FunctionTool.from_defaults(
fn=enhanced_web_search_and_update,
name="enhanced_web_search",
description="Search the web, extract content and images, and add them to the knowledge base for future queries."
)
def safe_import(module_name):
"""Safely import a module, return None if not available"""
try:
return __import__(module_name)
except ImportError:
return None
safe_globals = {
"__builtins__": {
"len": len, "str": str, "int": int, "float": float,
"list": list, "dict": dict, "sum": sum, "max": max, "min": min,
"round": round, "abs": abs, "sorted": sorted, "enumerate": enumerate,
"range": range, "zip": zip, "map": map, "filter": filter,
"any": any, "all": all, "type": type, "isinstance": isinstance,
"print": print, "open": open, "bool": bool, "set": set, "tuple": tuple
}
}
# Core modules (always available)
core_modules = [
"math", "datetime", "re", "os", "sys", "json", "csv", "random",
"itertools", "collections", "functools", "operator", "copy",
"decimal", "fractions", "uuid", "typing", "statistics", "pathlib",
"glob", "shutil", "tempfile", "pickle", "gzip", "zipfile", "tarfile",
"base64", "hashlib", "secrets", "hmac", "textwrap", "string",
"difflib", "socket", "ipaddress", "logging", "warnings", "traceback",
"pprint", "threading", "queue", "sqlite3", "urllib", "html", "xml",
"configparser"
]
for module in core_modules:
imported = safe_import(module)
if imported:
safe_globals[module] = imported
# Data science modules (may not be available)
optional_modules = {
"numpy": "numpy",
"np": "numpy",
"pandas": "pandas",
"pd": "pandas",
"scipy": "scipy",
"matplotlib": "matplotlib",
"plt": "matplotlib.pyplot",
"seaborn": "seaborn",
"sns": "seaborn",
"plotly": "plotly",
"sklearn": "sklearn",
"statsmodels": "statsmodels",
"PIL": "PIL",
"skimage": "skimage",
"pytz": "pytz",
"requests": "requests",
"bs4": "bs4",
"sympy": "sympy",
"tqdm": "tqdm",
"yaml": "yaml",
"toml": "toml"
}
for alias, module_name in optional_modules.items():
imported = safe_import(module_name)
if imported:
safe_globals[alias] = imported
# Special cases
if safe_globals.get("bs4"):
safe_globals["BeautifulSoup"] = safe_globals["bs4"].BeautifulSoup
if safe_globals.get("PIL"):
image_module = safe_import("PIL.Image")
if image_module:
safe_globals["Image"] = image_module
def execute_python_code(code: str) -> str:
try:
exec_locals = {}
exec(code, safe_globals, exec_locals)
if 'result' in exec_locals:
return str(exec_locals['result'])
else:
return "Code executed successfully"
except Exception as e:
return f"Code execution failed: {str(e)}"
code_execution_tool = FunctionTool.from_defaults(
fn=execute_python_code,
name="Python Code Execution",
description="Executes Python code safely for calculations and data processing"
)
def clean_response(response: str) -> str:
"""Clean response by removing common prefixes"""
response_clean = response.strip()
prefixes_to_remove = [
"FINAL ANSWER:", "Answer:", "The answer is:",
"Based on my analysis,", "After reviewing,",
"The result is:", "Final result:", "According to",
"In conclusion,", "Therefore,", "Thus,"
]
for prefix in prefixes_to_remove:
if response_clean.startswith(prefix):
response_clean = response_clean[len(prefix):].strip()
return response_clean
def llm_reformat(response: str, question: str) -> str:
"""Use LLM to reformat the response according to GAIA requirements"""
format_prompt = f"""Extract the exact answer from the response below. Follow GAIA formatting rules strictly.
GAIA Format Rules:
- ONLY the precise answer, no explanations
- No prefixes like "Answer:", "The result is:", etc.
- For numbers: just the number (e.g., "156", "3.14e+8")
- For names: just the name (e.g., "Martinez", "Sarah")
- For lists: comma-separated (e.g., "C++, Java, Python")
- For country codes: just the code (e.g., "FRA", "US")
- For yes/no: just "Yes" or "No"
Examples:
Question: "How many papers were published?"
Response: "The analysis shows 156 papers were published in total."
Answer: 156
Question: "What is the last name of the developer?"
Response: "The developer mentioned is Dr. Sarah Martinez from the AI team."
Answer: Martinez
Question: "List programming languages, alphabetized:"
Response: "The languages mentioned are Python, Java, and C++. Alphabetized: C++, Java, Python"
Answer: C++, Java, Python
Now extract the exact answer:
Question: {question}
Response: {response}
Answer:"""
try:
# Use the global LLM instance
formatting_response = proj_llm.complete(format_prompt)
answer = str(formatting_response).strip()
# Extract just the answer after "Answer:"
if "Answer:" in answer:
answer = answer.split("Answer:")[-1].strip()
return answer
except Exception as e:
print(f"LLM reformatting failed: {e}")
return response
def final_answer_tool(agent_response: str, question: str) -> str:
"""
Simplified final answer tool using only LLM reformatting.
Args:
agent_response: The raw response from agent reasoning
question: The original question for context
Returns:
Exact answer in GAIA format
"""
# Step 1: Clean the response
cleaned_response = clean_response(agent_response)
# Step 2: Use LLM reformatting
formatted_answer = llm_reformat(cleaned_response, question)
print(f"Original response cleaned: {cleaned_response[:100]}...")
print(f"LLM formatted answer: {formatted_answer}")
return formatted_answer
class EnhancedGAIAAgent:
def __init__(self):
print("Initializing Enhanced GAIA Agent...")
# Vérification du token HuggingFace
hf_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
if not hf_token:
print("Warning: HUGGINGFACEHUB_API_TOKEN not found, some features may not work")
# Initialize the dynamic query engine manager
self.dynamic_qe_manager = DynamicQueryEngineManager()
# Create enhanced agents with dynamic tools
self.external_knowledge_agent = ReActAgent(
name="external_knowledge_agent",
description="Advanced information retrieval with dynamic knowledge base",
system_prompt="""You are an advanced information specialist with a sophisticated RAG system.
Your knowledge base uses hybrid reranking and grows dynamically with each web search and document addition.
IMPORTANT INSTRUCTIONS FOR YOUR REASONING PROCESS:
1. Pay careful attention to ALL details in the user's question.
2. Think step by step about what is being asked, breaking down the requirements.
3. Identify specific qualifiers (e.g., "studio albums" vs just "albums", "between 2000-2010" vs "all time").
4. If searching for information, include ALL important details in your search query.
5. Double-check that your final answer addresses the EXACT question asked, not a simplified version.
For example:
- If asked "How many studio albums did Taylor Swift release between 2006-2010?", don't just search for
"Taylor Swift albums" - include "studio albums" AND the specific date range in your search.
- If asked about "Fortune 500 companies headquartered in California", don't just search for
"Fortune 500 companies" - include the location qualifier.
Always add relevant content to your knowledge base, then query it for answers.""",
tools=[
enhanced_web_search_tool,
self.dynamic_qe_manager.get_tool(),
code_execution_tool
],
llm=proj_llm,
max_steps=8,
verbose=True
)
self.code_agent = ReActAgent(
name="code_agent",
description="Handles Python code for calculations and data processing",
system_prompt="You are a Python programming specialist. You work with Python code to perform calculations, data analysis, and mathematical operations.",
tools=[code_execution_tool],
llm=code_llm,
max_steps=6,
verbose=True
)
# Fixed indentation: coordinator initialization inside __init__
self.coordinator = AgentWorkflow(
agents=[self.external_knowledge_agent, self.code_agent],
root_agent="external_knowledge_agent"
)
def download_gaia_file(self, task_id: str, api_url: str = "https://agents-course-unit4-scoring.hf.space") -> str:
"""Download file associated with task_id"""
try:
response = requests.get(f"{api_url}/files/{task_id}", timeout=30)
response.raise_for_status()
filename = f"task_{task_id}_file"
with open(filename, 'wb') as f:
f.write(response.content)
return filename
except Exception as e:
print(f"Failed to download file for task {task_id}: {e}")
return None
def add_documents_to_knowledge_base(self, file_path: str):
"""Add downloaded GAIA documents to the dynamic knowledge base"""
try:
documents = read_and_parse_content(file_path)
if documents:
self.dynamic_qe_manager.add_documents(documents)
print(f"Added {len(documents)} documents from {file_path} to dynamic knowledge base")
# Update the agent's tools with the refreshed query engine
self.external_knowledge_agent.tools = [
enhanced_web_search_tool,
self.dynamic_qe_manager.get_tool(), # Get the updated tool
code_execution_tool
]
return True
except Exception as e:
print(f"Failed to add documents from {file_path}: {e}")
return False
async def solve_gaia_question(self, question_data: Dict[str, Any]) -> str:
"""
Solve GAIA question with dynamic knowledge base integration
"""
question = question_data.get("Question", "")
task_id = question_data.get("task_id", "")
# Try to download and add file to knowledge base if task_id provided
file_path = None
if task_id:
try:
file_path = self.download_gaia_file(task_id)
if file_path:
# Add documents to dynamic knowledge base
self.add_documents_to_knowledge_base(file_path)
print(f"Successfully integrated GAIA file into dynamic knowledge base")
except Exception as e:
print(f"Failed to download/process file for task {task_id}: {e}")
# Enhanced context prompt with dynamic knowledge base awareness and step-by-step reasoning
context_prompt = f"""
GAIA Task ID: {task_id}
Question: {question}
{f'File processed and added to knowledge base: {file_path}' if file_path else 'No additional files'}
You are a general AI assistant. I will ask you a question.
IMPORTANT INSTRUCTIONS:
1. Think through this STEP BY STEP, carefully analyzing all aspects of the question.
2. Pay special attention to specific qualifiers like dates, types, categories, or locations.
3. Make sure your searches include ALL important details from the question.
4. Report your thoughts and reasoning process clearly.
5. Finish your answer with: FINAL ANSWER: [YOUR FINAL ANSWER]
YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
try:
ctx = Context(self.coordinator)
print("=== AGENT REASONING STEPS ===")
print(f"Dynamic knowledge base contains {len(self.dynamic_qe_manager.documents)} documents")
handler = self.coordinator.run(ctx=ctx, user_msg=context_prompt)
full_response = ""
async for event in handler.stream_events():
if isinstance(event, AgentStream):
print(event.delta, end="", flush=True)
full_response += event.delta
final_response = await handler
print("\n=== END REASONING ===")
# Extract the final formatted answer
#final_answer = final_answer_tool(str(final_response), question)
#print(f"Final GAIA formatted answer: {final_answer}")
#print(f"Knowledge base now contains {len(self.dynamic_qe_manager.documents)} documents")
return final_response
except Exception as e:
error_msg = f"Error processing question: {str(e)}"
print(error_msg)
return error_msg
def get_knowledge_base_stats(self):
"""Get statistics about the current knowledge base"""
return {
"total_documents": len(self.dynamic_qe_manager.documents),
"document_sources": [doc.metadata.get("source", "Unknown") for doc in self.dynamic_qe_manager.documents]
}
async def main():
res = enhanced_web_search_and_update("test")
agent = EnhancedGAIAAgent()
question_data = {
"Question": "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? List them !",
"task_id": ""
}
print(question_data)
answer = await agent.solve_gaia_question(question_data)
print(f"Answer: {answer}")
if __name__ == '__main__':
asyncio.run(main()) |