vikarti.anatra/Automated-AI-Web-Researcher-Ollama
1
0
Fork 0
mirror of https://github.com/TheBlewish/Automated-AI-Web-Researcher-Ollama.git synced 2025-01-19 00:47:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update Self_Improving_Search.py for windows

Browse source
This commit is contained in:
Hafeez 2024-11-20 21:46:03 +05:30 committed by GitHub
parent a53d85e284
commit aa73019d70
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
Self_Improving_Search.py
View file

@ -35,19 +35,20 @@ for name in ['root', 'duckduckgo_search', 'requests', 'urllib3']:
logging.getLogger(name).propagate = False logging.getLogger(name).propagate = False
class OutputRedirector: class OutputRedirector:
def __init__(self, stream=None): """Windows-compatible output redirection"""
self.stream = stream or StringIO() def __init__(self, stream=None):
self.original_stdout = sys.stdout self.stream = stream or StringIO()
self.original_stderr = sys.stderr self.original_stdout = sys.stdout
self.original_stderr = sys.stderr
def __enter__(self): def __enter__(self):
sys.stdout = self.stream sys.stdout = self.stream
sys.stderr = self.stream sys.stderr = self.stream
return self.stream return self.stream
def __exit__(self, exc_type, exc_val, exc_tb): def __exit__(self, exc_type, exc_val, exc_tb):
sys.stdout = self.original_stdout sys.stdout = self.original_stdout
sys.stderr = self.original_stderr sys.stderr = self.original_stderr
class EnhancedSelfImprovingSearch: class EnhancedSelfImprovingSearch:
def __init__(self, llm: LLMWrapper, parser: UltimateLLMResponseParser, max_attempts: int = 5): def __init__(self, llm: LLMWrapper, parser: UltimateLLMResponseParser, max_attempts: int = 5):