diff --git a/bin/simple_grammar.ml b/bin/simple_grammar.ml index 41d8295..70e39dc 100644 --- a/bin/simple_grammar.ml +++ b/bin/simple_grammar.ml @@ -141,16 +141,17 @@ let () = let url = ref "" in let help_str = "test" in let opts = [ - "-s", Arg.Set_string start, "startdir"; - "-n", Arg.Set_int item_count, "generate count items"; - "-p", Arg.Set_string prompt, "prompt filename"; - "-h", Arg.Set_string header_prompt, "header prompt filename"; - "-t", Arg.Set_string trailer_prompt, "trailer prompt filename"; - "-g", Arg.Set_string grammar, "grammar filename"; - "-x", Arg.Set_string suffix, "suffix"; - "-m", Arg.Set_string model, "model"; - "--llamacpp", Arg.Unit (fun () -> - lang_client := BLlamaCpp new Llama_cpp.llama_cpp_lang_model ; () + "-s", Arg.Set_string start, "startdir"; + "-w", Arg.Set_int window_size, "window_size"; + "-n", Arg.Set_int item_count, "generate count items"; + "-p", Arg.Set_string prompt, "prompt filename"; + "-h", Arg.Set_string header_prompt, "header prompt filename"; + "-t", Arg.Set_string trailer_prompt, "trailer prompt filename"; + "-g", Arg.Set_string grammar, "grammar filename"; + "-x", Arg.Set_string suffix, "suffix"; + "-m", Arg.Set_string model, "model"; + "--llamacpp", Arg.Unit (fun () -> + lang_client := BLlamaCpp new Llama_cpp.llama_cpp_lang_model ; () ), " Use Ollama (default)"; "-u", Arg.Set_string url, "url"; ] |> Arg.align in diff --git a/sim_dungeon_simple.sh b/sim_dungeon_simple.sh index a1252f0..607892a 100755 --- a/sim_dungeon_simple.sh +++ b/sim_dungeon_simple.sh @@ -17,12 +17,11 @@ The script is being used to generate and test various combinations of terms to c Here is the code we wrote so far " > $PROMPT_NAME2 #head -500 $PROMPT_NAME > $PROMPT_NAME2 cat $PROMPT_NAME >> $PROMPT_NAME2 -echo " ENDSRC . Please simulate its execution. Imagine running the code and all the output it would generate. then Imagine running that code. what would the game be like? Please rewrite my question if confusing. " >> $PROMPT_NAME2 +echo " ENDSRC . Please imagine the simulation its execution. Imagine running the code and all the output it would generate. Generate a grammar that we can use to generate prompts for create the dungeon using your knowledge. Write the grammar in EBNF notation. We will start with path through the dungeon starting at the entrance and going all the way to the ends." >> $PROMPT_NAME2 dune exec bin/simple.exe -- \ --openai -m "mixtral" -u "https://dev-hub.agentartificial.com" \ - -n 20 \ -f $PROMPT_NAME2 \ -s "data/dungeon/sim_dungeon_1_${DS}" \ -x ".txt" \ - -n 20 + -n 2000