Update main.py

main.py

@@ -6,42 +6,33 @@ import os
 import json
 import time
 import uuid
+import asyncio
 from typing import List, Dict, Any, Optional, AsyncGenerator
 
 # --- Configuration ---
 INFERENCE_API_KEY = os.environ.get("INFERENCE_API_KEY", "inference-00050468cc1c4a20bd5ca0997c752329")
 INFERENCE_API_URL = "https://api.inference.net/v1/chat/completions"
 SEARCH_API_URL = "https://rkihacker-brave.hf.space/search"
+NEWS_API_URL = "https://rkihacker-brave.hf.space/news" # Added News API URL
 MODEL_NAME = "Binglity-Lite"
 BACKEND_MODEL = "meta-llama/llama-3.1-8b-instruct/fp-8"
 
 # --- Final Advanced System Prompt ---
 SYSTEM_PROMPT = """
-You are "Binglity-Lite", a highly advanced AI search assistant. Your purpose is to provide users with accurate, comprehensive, and trustworthy answers by synthesizing information from a given set of web search results.
-
+You are "Binglity-Lite", a highly advanced AI search assistant. Your purpose is to provide users with accurate, comprehensive, and trustworthy answers by synthesizing information from a given set of web and news search results.
 **Core Directives:**
-
 1.  **Answer Directly**: Immediately address the user's question. **Do not** use introductory phrases like "Based on the search results..." or "Here is the information I found...". Your tone should be confident, objective, and encyclopedic.
-
 2.  **Synthesize, Don't Summarize**: Your primary task is to weave information from multiple sources into a single, cohesive, and well-structured answer. Do not simply describe what each source says one by one.
-
 3.  **Cite with Inline Markdown Links**: This is your most important instruction. When you present a fact or a piece of information from a source, you **must** cite it immediately using an inline Markdown link.
     *   **Format**: The format must be `[phrase or sentence containing the fact](URL)`. The URL must come from the `URL:` field of the provided source.
     *   **Example**: If a source with URL `https://example.com/science` says "The Earth is the third planet from the Sun", your output should be: "The Earth is the [third planet from the Sun](https://example.com/science)."
     *   **Rule**: Every piece of information in your answer must be attributable to a source via these inline links.
-
-4.  **Be Fact-Based**: Your entire response must be based **exclusively** on the information provided in the search results. Do not use any outside knowledge.
-
+4.  **Be Fact-Based**: Your entire response must be based **exclusively** on the information provided in the web and news search results. Do not use any outside knowledge.
 5.  **Filter for Relevance**: If a search result is not relevant to the user's query, ignore it completely. Do not mention it in your response.
-
 6.  **Handle Ambiguity**: If the search results are contradictory or insufficient to answer the question fully, state this clearly in your response, citing the conflicting sources.
-
 **Final Output Structure:**
-
 Your final response MUST be structured in two parts:
-
 1.  **The Synthesized Answer**: A well-written response that directly answers the user's query, with facts and statements properly cited using inline Markdown links as described above.
-
 2.  **Sources Section**: After the answer, add a section header `## Sources`. Under this header, provide a bulleted list of the full titles and URLs of every source you used.
     *   **Format**: `- [Title of Source](URL)`
 """
@@ -51,7 +42,7 @@ Your final response MUST be structured in two parts:
 app = FastAPI(
     title="Binglity-Lite API",
     description="A web search-powered, streaming-capable chat completions API.",
-    version="1.2.0",
+    version="1.3.0", # Version updated
 )
 
 # --- Pydantic Models for OpenAI Compatibility ---
@@ -66,7 +57,7 @@ class ChatCompletionRequest(BaseModel):
     temperature: Optional[float] = 0.7
     stream: Optional[bool] = False
 
-# --- Web Search Function ---
+# --- Web Search Functions ---
 async def perform_web_search(query: str) -> List[Dict[str, Any]]:
     async with httpx.AsyncClient() as client:
         try:
@@ -75,7 +66,11 @@ async def perform_web_search(query: str) -> List[Dict[str, Any]]:
                 params={"query": query, "max_results": 10}
             )
             response.raise_for_status()
-            return response.json()
+            results = response.json()
+            # Add source type to each result
+            for result in results:
+                result['source_type'] = 'Web'
+            return results
         except httpx.HTTPStatusError as e:
             print(f"Error from search API: {e.response.text}")
             return []
@@ -83,13 +78,38 @@ async def perform_web_search(query: str) -> List[Dict[str, Any]]:
             print(f"An unexpected error occurred during web search: {str(e)}")
             return []
 
+async def perform_news_search(query: str) -> List[Dict[str, Any]]:
+    """Performs a search against the news API."""
+    async with httpx.AsyncClient() as client:
+        try:
+            # Parameters can be adjusted as needed, e.g., region
+            response = await client.get(
+                NEWS_API_URL,
+                params={"query": query, "max_results": 10, "region": "en-US"}
+            )
+            response.raise_for_status()
+            results = response.json()
+            # Add source type to each result
+            for result in results:
+                result['source_type'] = 'News'
+            return results
+        except httpx.HTTPStatusError as e:
+            print(f"Error from news API: {e.response.text}")
+            return []
+        except Exception as e:
+            print(f"An unexpected error occurred during news search: {str(e)}")
+            return []
+
+
 def format_search_results_for_prompt(results: List[Dict[str, Any]]) -> str:
+    """Formats combined search results for the language model prompt."""
     if not results:
         return "No relevant search results were found. Inform the user that you could not find information on their query."
     
-    formatted = "### Web Search Results ###\n\n"
+    formatted = "### Search Results ###\n\n"
     for i, result in enumerate(results):
-        formatted += f"Source [{i+1}]:\n"
+        source_type = result.get('source_type', 'Search') # Default in case it's missing
+        formatted += f"Source [{i+1}] ({source_type}):\n"
         formatted += f"Title: {result.get('title', 'N/A')}\n"
         formatted += f"URL: {result.get('url', 'N/A')}\n"
         formatted += f"Content: {result.get('description', 'N/A')}\n\n"
@@ -139,10 +159,24 @@ async def chat_completions(request: ChatCompletionRequest):
     if not user_query or request.messages[-1].role.lower() != 'user':
         raise HTTPException(status_code=400, detail="The last message must be from the 'user' and contain content.")
 
-    search_results = await perform_web_search(user_query)
-    formatted_results = format_search_results_for_prompt(search_results)
+    # Perform web and news searches concurrently
+    web_results, news_results = await asyncio.gather(
+        perform_web_search(user_query),
+        perform_news_search(user_query)
+    )
+
+    # Combine results and remove duplicates by URL
+    combined_results = []
+    seen_urls = set()
+    for result in web_results + news_results:
+        url = result.get('url')
+        if url and url not in seen_urls:
+            combined_results.append(result)
+            seen_urls.add(url)
+    
+    formatted_results = format_search_results_for_prompt(combined_results)
     
-    final_user_prompt = f"User's question: \"{user_query}\"\n\nUse the web search results below to answer the user's question. Follow all rules in your system prompt exactly.\n\n{formatted_results}"
+    final_user_prompt = f"User's question: \"{user_query}\"\n\nUse the web and news search results below to answer the user's question. Follow all rules in your system prompt exactly.\n\n{formatted_results}"
     
     payload = {
         "model": BACKEND_MODEL,