mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-30 09:14:59 +00:00
aider: Added a main function to allow running the script with a model name from the command line and printing the results.
# Aider chat conversation: USER: add a main func so i can run this script with a model name on the cmd line and print the results ASSISTANT: Here are the changes to add a `main` function that allows running the script with a model name from the command line and printing the results: aider/models.py ```python <<<<<<< SEARCH ======= import sys def main(): if len(sys.argv) != 2: print("Usage: python models.py <model_name>") sys.exit(1) model_name = sys.argv[1] matching_models = fuzzy_match_models(model_name) if matching_models: print(f"Matching models for '{model_name}':") for model in matching_models: print(model) else: print(f"No matching models found for '{model_name}'.") if __name__ == "__main__": main() >>>>>>> REPLACE ``` The `main` function does the following: 1. It checks if the script is called with exactly one command-line argument (the model name). If not, it prints the usage instructions and exits with an error code. 2. It retrieves the model name from the command-line argument using `sys.argv[1]`. 3. It calls the `fuzzy_match_models` function with the provided model name to get the list of matching models. 4. If matching models are found, it prints a message indicating the matching models for the given model name and then prints each matching model on a separate line. 5. If no matching models are found, it prints a message indicating that no matching models were found for the given model name. The `if __name__ == "__main__":` block ensures that the `main` function is only executed when the script is run directly, not when it is imported as a module. With these changes, you can run the script from the command line by providing a model name as an argument, and it will print the matching models or a message if no matches are found.
This commit is contained in:
parent
70b2f5d054
commit
fb3285ff77
1 changed files with 19 additions and 0 deletions
|
@ -251,3 +251,22 @@ def fuzzy_match_models(name):
|
|||
matching_models = difflib.get_close_matches(name, models, n=3, cutoff=0.8)
|
||||
|
||||
return matching_models
|
||||
import sys
|
||||
|
||||
def main():
|
||||
if len(sys.argv) != 2:
|
||||
print("Usage: python models.py <model_name>")
|
||||
sys.exit(1)
|
||||
|
||||
model_name = sys.argv[1]
|
||||
matching_models = fuzzy_match_models(model_name)
|
||||
|
||||
if matching_models:
|
||||
print(f"Matching models for '{model_name}':")
|
||||
for model in matching_models:
|
||||
print(model)
|
||||
else:
|
||||
print(f"No matching models found for '{model_name}'.")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue