jarianc/reddit-cli
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: Update AI endpoint to use home.ms:4000 with API key 111

Browse Source
This commit is contained in:
Your Name 2026-02-10 23:25:21 -06:00
parent e3bd8f66ac
commit 079d3dd899
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/mcp_server.py
View File

@ -25,7 +25,7 @@ app = Flask(__name__)
CORS(app) CORS(app)
# Global configuration # Global configuration
OLLAMA_BASE_URL = os.getenv("OLLAMA_BASE_URL", "http://192.168.8.223:11434") OLLAMA_BASE_URL = os.getenv("OLLAMA_BASE_URL", "http://home.ms:4000")
OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "gpt-oss:20b") OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "gpt-oss:20b")
REDDIT_BASE_URL = "https://www.reddit.com" REDDIT_BASE_URL = "https://www.reddit.com"
@ -173,12 +173,14 @@ class RedditClient:
class AIClient: class AIClient:
"""Client for interacting with Ollama AI API with retry logic""" """Client for interacting with AI API with retry logic"""
def __init__(self): def __init__(self):
self.base_url = OLLAMA_BASE_URL self.base_url = OLLAMA_BASE_URL
self.model = OLLAMA_MODEL self.model = OLLAMA_MODEL
self.session = requests.Session() self.session = requests.Session()
# Set the API key for the new endpoint
self.api_key = "111" # As specified in the feedback
def _make_request_with_retry(self, url: str, json_data: Dict = None, max_retries: int = 3, def _make_request_with_retry(self, url: str, json_data: Dict = None, max_retries: int = 3,
retry_delay: float = 1.0) -> requests.Response: retry_delay: float = 1.0) -> requests.Response:
@ -199,7 +201,9 @@ class AIClient:
""" """
for attempt in range(max_retries + 1): for attempt in range(max_retries + 1):
try: try:
response = self.session.post(url, json=json_data, timeout=30) # Add API key to headers
headers = {"x-api-key": self.api_key}
response = self.session.post(url, json=json_data, headers=headers, timeout=30)
response.raise_for_status() response.raise_for_status()
return response return response
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e: