Spaces:
Running
Running
milwright
commited on
Commit
·
69cb0c3
1
Parent(s):
f92783b
fix preview to match space_template: add dynamic url extraction and api headers
Browse files
app.py
CHANGED
@@ -567,6 +567,20 @@ class SpaceGenerator:
|
|
567 |
except:
|
568 |
pass
|
569 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
570 |
# Build messages for API with grounding context in system prompt
|
571 |
system_content = config.get('system_prompt', 'You are a helpful AI assistant.')
|
572 |
language = config.get('language', 'English')
|
@@ -603,6 +617,13 @@ class SpaceGenerator:
|
|
603 |
"stream": False
|
604 |
}
|
605 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
606 |
response = requests.post(
|
607 |
"https://openrouter.ai/api/v1/chat/completions",
|
608 |
headers=headers,
|
|
|
567 |
except:
|
568 |
pass
|
569 |
|
570 |
+
# Check for dynamic URLs in message if enabled
|
571 |
+
if config.get('enable_dynamic_urls', True):
|
572 |
+
urls_in_message = extract_urls_from_text(message)
|
573 |
+
if urls_in_message:
|
574 |
+
dynamic_context = "\n📎 **Dynamic Context:**\n"
|
575 |
+
for url in urls_in_message[:3]: # Limit to 3 URLs
|
576 |
+
try:
|
577 |
+
content = fetch_url_content(url, max_chars=2500)
|
578 |
+
if not content.startswith("❌"):
|
579 |
+
dynamic_context += f"\n{content}"
|
580 |
+
except:
|
581 |
+
pass
|
582 |
+
grounding_context += dynamic_context
|
583 |
+
|
584 |
# Build messages for API with grounding context in system prompt
|
585 |
system_content = config.get('system_prompt', 'You are a helpful AI assistant.')
|
586 |
language = config.get('language', 'English')
|
|
|
617 |
"stream": False
|
618 |
}
|
619 |
|
620 |
+
# Add additional headers to match space template
|
621 |
+
space_id = os.environ.get('SPACE_ID', '')
|
622 |
+
headers.update({
|
623 |
+
"HTTP-Referer": f"https://huggingface.co/spaces/{space_id}" if space_id else "https://huggingface.co",
|
624 |
+
"X-Title": config.get('name', 'AI Assistant')
|
625 |
+
})
|
626 |
+
|
627 |
response = requests.post(
|
628 |
"https://openrouter.ai/api/v1/chat/completions",
|
629 |
headers=headers,
|