diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/umapyai/__init__.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/umapyai/__init__.py b/src/umapyai/__init__.py index ce2099f..2aab5fe 100644 --- a/src/umapyai/__init__.py +++ b/src/umapyai/__init__.py @@ -30,7 +30,8 @@ def prompt(rag_context, user_query, is_first_turn=True): system_prompt = ( 'You are a friendly and expert "Uma Musume: Pretty Derby" build guide advisor. ' 'Your personality is that of a helpful stable master, guiding a new trainer. ' - 'Your goal is to provide a comprehensive and encouraging answer to the user\'s question.' + 'Your goal is to provide a comprehensive and encouraging answer to the user\'s question. ' + 'You are a fan-made guide and not affiliated with Cygames or any other company.' ) instruction = ( "Carefully analyse the user's question and the provided context. " @@ -43,7 +44,7 @@ def prompt(rag_context, user_query, is_first_turn=True): f"{system_prompt}\n\n" f"## Instruction\n{instruction}\n\n" "## Context Provided\n" - f"--- START OF CONTEXT ---\n{rag_context}\n--- END OF CONTEXT---\n\n" + f"--- START OF CONTEXT ---\n{rag_context}\n--- END OF CONTEXT ---\n\n" f"## User's Question\n{user_query}\n\n" "## Your Answer") else: @@ -51,7 +52,7 @@ def prompt(rag_context, user_query, is_first_turn=True): "Here is some new information that might be relevant to your follow-up question. " "Please synthesise it with our ongoing conversation to provide your answer.\n\n" "## Additional Context\n" - f"--- START OF CONTEXT ---\n{rag_context}\n--- END OF CONTEXT---\n\n" + f"--- START OF CONTEXT ---\n{rag_context}\n--- END OF CONTEXT ---\n\n" f"## User's Question\n{user_query}\n\n" "## Your Answer") |