diff --git a/src/app_context.sh b/src/app_context.sh new file mode 100644 index 0000000..f5f6023 --- /dev/null +++ b/src/app_context.sh @@ -0,0 +1,15 @@ +#!/usr/bin/env sh +# shellcheck enable=all + +# Pull in context from mail app + +# Pull in context from calendar app + +# Pull in context from notes app + +# Pull in context from reminders app + +# Pull in context from terminal app + +# Pull in context from safari app + diff --git a/src/llm_session_management.sh b/src/llm_session_management.sh index a18a3dd..b5d20a0 100644 --- a/src/llm_session_management.sh +++ b/src/llm_session_management.sh @@ -15,6 +15,14 @@ generate_prompt() { add_system_info=$8 add_directory_info=$9 add_clipboard_info=${10} + add_screenshot_info=${11} + add_text_info=${12} + add_mail_info=${13} + add_calendar_info=${14} + add_notes_info=${15} + add_reminders_info=${16} + add_terminal_info=${17} + add_safari_info=${18} # Check that inputs are valid check_mode mode || return 1 diff --git a/src/main.sh b/src/main.sh index c771a46..19869b7 100755 --- a/src/main.sh +++ b/src/main.sh @@ -40,6 +40,7 @@ esac . "${DIR}/tokens.sh" . "${DIR}/content_processing.sh" . "${DIR}/prompt_generators.sh" +. "${DIR}/app_context.sh" . "${DIR}/llm_session_management.sh" # Write a settings file if there isn't one already @@ -55,6 +56,8 @@ search_term_list="" surf_and_add_results=false add_directory_info=false add_system_info=false add_clipboard_info=false add_usage_info=false task_model_override=false casual_model_override=false balanced_model_override=false serious_model_override=false +add_screenshot_info=false add_text_info=false +add_mail_info=false add_calendar_info=false add_notes_info=false add_reminders_info=false add_terminal_info=false add_safari_info=false # Make verbose a global variable VERBOSE=false @@ -206,6 +209,30 @@ while [ $# -gt 0 ]; do surf_and_add_results=true ;; + # Add screenshot info + -sc|--screenshot) add_screenshot_info=true ;; + + # Add text info + -t|--text) add_text_info=true ;; + + # Pull in context from mail app + --mail) add_mail_info=true ;; + + # Pull in context from calendar app + --calendar) add_calendar_info=true ;; + + # Pull in context from notes app + --notes) add_notes_info=true ;; + + # Pull in context from reminders app + --reminders) add_reminders_info=true ;; + + # Pull in context from terminal app + --terminal) add_terminal_info=true ;; + + # Pull in context from safari app + --safari) add_safari_info=true ;; + # Add system information to the context -y | --system) add_system_info=true ;; @@ -299,7 +326,15 @@ prompt=$( "${add_usage_info}" \ "${add_system_info}" \ "${add_directory_info}" \ - "${add_clipboard_info}" + "${add_clipboard_info}" \ + "${add_screenshot_info}" \ + "${add_text_info}" \ + "${add_mail_info}" \ + "${add_calendar_info}" \ + "${add_notes_info}" \ + "${add_reminders_info}" \ + "${add_terminal_info}" \ + "${add_safari_info}" ) || { echo "Error: Failed to generate prompt." >&2 exit 1