qwen_added

Author: RobuRishabh

Assigment - level 2.pdf

@@ -33,13 +33,13 @@ def search_flights(origin=None, destination=None, flight_number=None, airline=No
     """
     print(f"🔍 Searching for: Origin={origin}, Destination={destination}, Flight Number={flight_number}, Airline={airline}")
 
-    # ✅ If flight number is provided, prioritize searching by flight number only
+    # If flight number is provided, prioritize searching by flight number only
     if flight_number:
         matches = [flight for flight in flights if flight["flight_number"].lower() == flight_number.lower()]
         print(f"🔍 Flight number search results: {matches}")
         return matches
 
-    # ✅ If no flight number, apply standard search
+    # If no flight number, apply standard search
     if not any([origin, destination, airline]):
         print("⚠️ No valid search parameters provided. Returning an empty list.")
         return []

mock_database.py

@@ -7,7 +7,7 @@
 
 load_dotenv()
 OLLAMA_URL = os.getenv("OLLAMA_URL", "http://localhost:11434")
-OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama2:latest")
+OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "qwen2.5-coder:3b")
 
 def initialize_ollama():
     try:

ollama_api.py

@@ -7,7 +7,7 @@
 
 # Load environment variables
 load_dotenv()
-OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama2:latest")
+OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "qwen2.5-coder:3b")
 OLLAMA_AVAILABLE = check_ollama_availability()
 
 # Initialize Ollama LLM
@@ -69,13 +69,13 @@ def extract_entities_ollama(query):
             json_str = json_match.group(0)
             extracted = json.loads(json_str)
 
-            # ✅ Clean extracted values
+            # Clean extracted values
             if extracted.get("destination") == "City Name":
                 extracted["destination"] = None  # Ignore placeholder values
             if extracted.get("origin") == "NYC":
-                extracted["origin"] = "New York"  # Fix Ollama misinterpretation
+                extracted["origin"] = "New York"  # Map city codes to full names
 
-            # ✅ If no flight number is extracted, try regex
+            # If no flight number is extracted, try regex
             if not extracted.get("flight_number"):
                 extracted["flight_number"] = extract_flight_number(query)
 

query_handler.py

@@ -9,7 +9,7 @@
 def mock_env():
     original_env = os.environ.copy()
     os.environ["OLLAMA_URL"] = "http://test:11434"
-    os.environ["OLLAMA_MODEL"] = "llama2:test"
+    os.environ["OLLAMA_MODEL"] = "qwen2.5-coder:3b"
     yield
     os.environ.clear()
     os.environ.update(original_env)

tests/test_ollama_api.py

@@ -11,7 +11,7 @@
 @pytest.fixture
 def mock_env():
     original_env = os.environ.copy()
-    os.environ["OLLAMA_MODEL"] = "llama2:test"
+    os.environ["OLLAMA_MODEL"] = "qwen2.5-coder:3b"
     yield
     os.environ.clear()
     os.environ.update(original_env)