Reorganize to hide internals, and add start.sh

This commit is contained in:
Burke Johnson 2024-11-25 22:32:21 -06:00
parent 529de5d259
commit 18342fda9e
11 changed files with 25 additions and 12 deletions

View file

@ -6,10 +6,10 @@ from colorama import Fore, Style
import logging
import sys
from io import StringIO
from web_scraper import get_web_content, can_fetch
from llm_config import get_llm_config
from llm_response_parser import UltimateLLMResponseParser
from llm_wrapper import LLMWrapper
from .web_scraper import get_web_content, can_fetch
from .llm_config import get_llm_config
from .llm_response_parser import UltimateLLMResponseParser
from .llm_wrapper import LLMWrapper
from urllib.parse import urlparse
# Set up logging

0
src/__init__.py Normal file
View file

View file

@ -4,12 +4,12 @@ from colorama import init, Fore, Style
import logging
import time
from io import StringIO
from Self_Improving_Search import EnhancedSelfImprovingSearch
from llm_config import get_llm_config
from llm_response_parser import UltimateLLMResponseParser
from llm_wrapper import LLMWrapper
from strategic_analysis_parser import StrategicAnalysisParser
from research_manager import ResearchManager
from .Self_Improving_Search import EnhancedSelfImprovingSearch
from .llm_config import get_llm_config
from .llm_response_parser import UltimateLLMResponseParser
from .llm_wrapper import LLMWrapper
from .strategic_analysis_parser import StrategicAnalysisParser
from .research_manager import ResearchManager
# Initialize colorama
if os.name == 'nt': # Windows-specific initialization

View file

@ -2,7 +2,7 @@ import re
from typing import Dict, List, Union, Optional
import logging
import json
from strategic_analysis_parser import StrategicAnalysisParser, AnalysisResult, ResearchFocus
from .strategic_analysis_parser import StrategicAnalysisParser, AnalysisResult, ResearchFocus
# Set up logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

View file

@ -2,7 +2,7 @@ import os
from llama_cpp import Llama
import requests
import json
from llm_config import get_llm_config
from .llm_config import get_llm_config
from openai import OpenAI
from anthropic import Anthropic

13
start.sh Executable file
View file

@ -0,0 +1,13 @@
#!/bin/bash
if [ ! -d "venv" ]; then
python3 -m venv venv
fi
source venv/bin/activate
pip install -r "src/requirements.txt"
python -m src
deactivate