Update typehints

Author: tammy-baylis-swi

instrumentation/opentelemetry-instrumentation-psycopg/src/opentelemetry/instrumentation/psycopg/__init__.py

@@ -109,6 +109,9 @@
 from psycopg import (
     AsyncCursor as pg_async_cursor,  # pylint: disable=import-self,no-name-in-module
 )
+from psycopg import (
+    Connection as pg_connection,  # pylint: disable=no-name-in-module
+)
 from psycopg import (
     Cursor as pg_cursor,  # pylint: disable=no-name-in-module,import-self
 )
@@ -197,8 +200,8 @@ def _uninstrument(self, **kwargs):
     # TODO(owais): check if core dbapi can do this for all dbapi implementations e.g, pymysql and mysql
     @staticmethod
     def instrument_connection(
-        connection,
-        tracer_provider=None,
+        connection: pg_connection,
+        tracer_provider: typing.Optional[trace_api.TracerProvider] = None,
         enable_commenter: bool = False,
         commenter_options: dict = None,
     ):

instrumentation/opentelemetry-instrumentation-psycopg2/src/opentelemetry/instrumentation/psycopg2/__init__.py

@@ -106,6 +106,9 @@
 from typing import Collection
 
 import psycopg2
+from psycopg2.extensions import (
+    connection as pg_connection,  # pylint: disable=no-name-in-module
+)
 from psycopg2.extensions import (
     cursor as pg_cursor,  # pylint: disable=no-name-in-module
 )
@@ -161,8 +164,8 @@ def _uninstrument(self, **kwargs):
     # TODO(owais): check if core dbapi can do this for all dbapi implementations e.g, pymysql and mysql
     @staticmethod
     def instrument_connection(
-        connection,
-        tracer_provider=None,
+        connection: pg_connection,
+        tracer_provider: typing.Optional[trace_api.TracerProvider] = None,
         enable_commenter: bool = False,
         commenter_options: dict = None,
     ):