| """ |
| Query analyzer for code search intent. |
| """ |
|
|
| from typing import Any |
|
|
| from .dictionaries.frameworks import FRAMEWORKS |
| from .dictionaries.languages import PROGRAMMING_LANGUAGES |
|
|
| CODE_KEYWORDS = [ |
| "function", |
| "class", |
| "method", |
| "algorithm", |
| "implementation", |
| "example", |
| "tutorial", |
| "pattern", |
| "template", |
| "snippet", |
| "code", |
| "programming", |
| "development", |
| "api", |
| "library", |
| ] |
|
|
|
|
| def analyze_code_query(query: str) -> dict[str, Any]: |
| """ |
| Analyze a query to determine if it's code-related and extract relevant information. |
| |
| Args: |
| query: Search query to analyze |
| |
| Returns: |
| Analysis results with query classification and extracted info |
| """ |
| query_lower = query.lower() |
|
|
| detected_languages = [lang for lang in PROGRAMMING_LANGUAGES if lang in query_lower] |
| detected_frameworks = [fw for fw in FRAMEWORKS if fw in query_lower] |
| code_indicators = [kw for kw in CODE_KEYWORDS if kw in query_lower] |
|
|
| is_code_query = len(detected_languages) > 0 or len(detected_frameworks) > 0 or len(code_indicators) > 0 |
|
|
| return { |
| "is_code_query": is_code_query, |
| "confidence": min( |
| 1.0, |
| (len(detected_languages) + len(detected_frameworks) + len(code_indicators)) * 0.3, |
| ), |
| "languages": detected_languages, |
| "frameworks": detected_frameworks, |
| "code_indicators": code_indicators, |
| "enhanced_query_recommended": is_code_query, |
| } |
|
|
|
|
| def analyze_query_for_code_search(query: str) -> dict[str, Any]: |
| """ |
| Standalone function to analyze if a query is code-related. |
| """ |
| return analyze_code_query(query) |
|
|
