SYMBOL INDEX (69 symbols across 15 files)

FILE: gradio/app.py
  function solve_problem (line 28) | def solve_problem(problem, max_iterations, reward_threshold, swift_model...

FILE: swiftsage/agents/agent.py
  class Agent (line 4) | class Agent(ABC):
    method __init__ (line 5) | def __init__(self, prompt_template, llm_client):
    method generate_response (line 10) | def generate_response(self, prompt):

FILE: swiftsage/agents/feedback_agent.py
  class Feedback (line 10) | class Feedback:
    method __init__ (line 11) | def __init__(self, prompt_template, llm_client, K=2):
    method calculate_reward (line 19) | def calculate_reward(self, problem, reasoning, current_solution, prefi...
    method should_consult_sage (line 56) | def should_consult_sage(self):

FILE: swiftsage/agents/retrieval_agent.py
  class RetrievalAugmentation (line 4) | class RetrievalAugmentation:
    method __init__ (line 6) | def __init__(self, dataset, embeddings):
    method get_similar_examples (line 10) | def get_similar_examples(self, query_embedding, n=3):

FILE: swiftsage/agents/sage_agent.py
  class SageAgent (line 13) | class SageAgent(Agent):
    method __init__ (line 14) | def __init__(self, prompt_template, llm_client):
    method generate_response (line 20) | def generate_response(self, prompt, reasoning, current_solution, prefi...

FILE: swiftsage/agents/swift_agent.py
  class SwiftAgent (line 13) | class SwiftAgent(Agent):
    method __init__ (line 14) | def __init__(self, prompt_template, llm_client, retrieval_augmentation...
    method generate_response (line 20) | def generate_response(self, prompt, reasoning, current_solution, plan,...
    method get_query_embedding (line 60) | def get_query_embedding(self, query):

FILE: swiftsage/agents/swiftsage.py
  class SwiftSage (line 11) | class SwiftSage:
    method __init__ (line 12) | def __init__(self, dataset, embeddings, prompt_template_dir, swift_con...
    method solve (line 28) | def solve(self, problem, max_iterations=10, reward_threshold=8):

FILE: swiftsage/benchmark/data_loader.py
  function load_jsonl (line 16) | def load_jsonl(file):
  function load_data (line 26) | def load_data(

FILE: swiftsage/benchmark/data_utils.py
  function lower_keys (line 12) | def lower_keys(example):
  function _fix_fracs (line 23) | def _fix_fracs(string):
  function _fix_a_slash_b (line 55) | def _fix_a_slash_b(string):
  function _fix_sqrt (line 72) | def _fix_sqrt(string):
  function convert_word_number (line 77) | def convert_word_number(text:str) -> str:
  function strip_string (line 106) | def strip_string(string):
  function extract_multi_choice_answer (line 233) | def extract_multi_choice_answer(pred_str):
  function extract_answer (line 244) | def extract_answer(pred_str, data_name):
  function parse_ground_truth (line 299) | def parse_ground_truth(example: Dict[str, Any], data_name):
  function parse_question (line 317) | def parse_question(example, data_name):
  function _test_extract_answer (line 342) | def _test_extract_answer():

FILE: swiftsage/benchmark/evaluate.py
  function evaluate (line 15) | def evaluate(samples: list=None, file_path: str=None):

FILE: swiftsage/benchmark/grader.py
  function parse_digits (line 21) | def parse_digits(num):
  function is_digit (line 36) | def is_digit(num):
  function str_to_pmatrix (line 41) | def str_to_pmatrix(input_str):
  function math_equal (line 54) | def math_equal(prediction: Union[bool, float, str],
  function math_equal_process (line 171) | def math_equal_process(param):
  function numeric_equal (line 175) | def numeric_equal(prediction: float, reference: float):
  function symbolic_equal (line 185) | def symbolic_equal(a, b):
  function symbolic_equal_process (line 242) | def symbolic_equal_process(a, b, output_queue):
  function call_with_timeout (line 247) | def call_with_timeout(func, *args, timeout=1, **kwargs):
  function _test_math_equal (line 262) | def _test_math_equal():

FILE: swiftsage/cli.py
  function run_test (line 13) | def run_test(swiftsage, problem, max_iterations=5, reward_threshold=8):
  function parse_args (line 21) | def parse_args():
  function main (line 56) | def main():

FILE: swiftsage/evaluate_benchmark.py
  function run_benchmark (line 20) | def run_benchmark(swiftsage, args, max_iterations=5, reward_threshold=8):
  function main (line 66) | def main(args):

FILE: swiftsage/utils/code_executor.py
  class GenericRuntime (line 14) | class GenericRuntime:
    method __init__ (line 19) | def __init__(self):
    method exec_code (line 26) | def exec_code(self, code_piece: str) -> None:
    method eval_code (line 29) | def eval_code(self, expr: str) -> any:
    method inject (line 32) | def inject(self, var_dict):
    method answer (line 36) | def answer(self):
  class PythonExecutor (line 40) | class PythonExecutor:
    method __init__ (line 41) | def __init__(
    method execute (line 55) | def execute(self, code):
    method apply (line 81) | def apply(self, code):

FILE: swiftsage/utils/commons.py
  function setup_logging (line 36) | def setup_logging():
  function extract_and_parse_markup (line 60) | def extract_and_parse_markup(text):
  class PromptTemplate (line 93) | class PromptTemplate:
    method __init__ (line 94) | def __init__(self, template_dir):
    method load_templates (line 99) | def load_templates(self):
    method format (line 105) | def format(self, key, **kwargs):
  class LLMClient (line 112) | class LLMClient:
    method __init__ (line 113) | def __init__(self, model_id, api_config, temperature=0.3, top_p=1.0, m...
    method generate_response (line 129) | def generate_response(self, messages):